Merge pull request #123 from sdcb/pr-120

merge BaseModel into SegModel

Author: sdcb

src/Sdcb.PaddleSeg/BaseModel.cs

@@ -2,10 +2,32 @@
 
 namespace Sdcb.PaddleSeg
 {
-    public abstract class SegModel : BaseModel
+    public abstract class SegModel
     {
+        /// <summary>
+        /// Create the paddle config for the model.
+        /// </summary>
+        /// <returns>The paddle config for model.</returns>
+        public abstract PaddleConfig CreateConfig();
+
+        /// <summary>
+        /// Configure the device related config of the <see cref="PaddleConfig"/>.
+        /// </summary>
+        /// <param name="configure">The device and configure of the <see cref="PaddleConfig"/>, pass null to using model's <see cref="DefaultDevice"/>.</param>
+        /// <param name="config">The PaddleConfig to modify.</param>
+        public virtual void ConfigureDevice(PaddleConfig config, Action<PaddleConfig>? configure = null)
+        {
+            config.Apply(configure ?? DefaultDevice);
+        }
+
+        /// <summary>
+        /// Create a <see cref="SegModel"/> from the directory path.
+        /// </summary>
         public static SegModel FromDirectory(string directoryPath) => new FileSegModel(directoryPath);
 
-        public override Action<PaddleConfig> DefaultDevice => PaddleDevice.Mkldnn();
+        /// <summary>
+        /// Default device configuration for the model.
+        /// </summary>
+        public virtual Action<PaddleConfig> DefaultDevice => PaddleDevice.Mkldnn();
     }
 }

src/Sdcb.PaddleSeg/SegModel.cs

@@ -26,7 +26,7 @@ public void LocalV3TableTest(string modelName)
     [Theory]
     [InlineData("en_ppstructure_mobile_v2.0_SLANet", "<table><thead><tr><td>Methods</td>")]
     [InlineData("ch_ppstructure_mobile_v2.0_SLANet", "<table><tbody><tr><td>Methods</td>")]
-    public async void LocalV3TableRebuild(string modelName, string expectedHtmlStart)
+    public async Task LocalV3TableRebuild(string modelName, string expectedHtmlStart)
     {
         using Mat src = Cv2.ImRead("samples/table.jpg");
         Task<TableDetectionResult> tableResultTask = Task.Run(() =>