reshape primary input for paddle ocr table model

sdcb · sdcb · commit a1e99c578c21 · 2024-06-04T18:36:35.000+08:00
diff --git a/projects/PaddleOCR/Sdcb.OpenVINO.PaddleOCR/Models/TableRecognitionModel.cs b/projects/PaddleOCR/Sdcb.OpenVINO.PaddleOCR/Models/TableRecognitionModel.cs
@@ -48,6 +48,12 @@ internal TableRecognitionModel(IReadOnlyList<string> labels, bool mergeNoSpanStr
         }
     }
 
+    /// <inheritdoc/>
+    public override void AfterReadModel(Model model)
+    {
+        model.ReshapePrimaryInput(new PartialShape(1, 3, 488, 488));
+    }
+
     /// <summary>
     /// Creates a new instance of the <see cref="FileTableRecognizationModel"/> class from a specified directory path and label path.
     /// </summary>

Original file line number	Diff line number	Diff line change
`@@ -48,6 +48,12 @@ internal TableRecognitionModel(IReadOnlyList<string> labels, bool mergeNoSpanStr`
`48`	`48`	`}`
`49`	`49`	`}`
`50`	`50`
	`51`	`+ /// <inheritdoc/>`
	`52`	`+ public override void AfterReadModel(Model model)`
	`53`	`+ {`
	`54`	`+ model.ReshapePrimaryInput(new PartialShape(1, 3, 488, 488));`
	`55`	`+ }`
	`56`	`+`
`51`	`57`	`/// <summary>`
`52`	`58`	`/// Creates a new instance of the <see cref="FileTableRecognizationModel"/> class from a specified directory path and label path.`
`53`	`59`	`/// </summary>`