Jelajahi Sumber

update model

xujiayue 2 tahun lalu
induk
melakukan
e0ccdf7550

TEMPAT SAMPAH
SLAnet_1004/inference.pdiparams


TEMPAT SAMPAH
SLAnet_1004/inference.pdiparams.info


TEMPAT SAMPAH
SLAnet_1004/inference.pdmodel


TEMPAT SAMPAH
SLAnet_1011/inference.pdiparams


TEMPAT SAMPAH
SLAnet_1011/inference.pdiparams.info


TEMPAT SAMPAH
SLAnet_1011/inference.pdmodel


TEMPAT SAMPAH
det_table_1012/inference.pdiparams


TEMPAT SAMPAH
det_table_1012/inference.pdiparams.info


TEMPAT SAMPAH
det_table_1012/inference.pdmodel


TEMPAT SAMPAH
det_table_1014/inference.pdiparams


TEMPAT SAMPAH
det_table_1014/inference.pdiparams.info


TEMPAT SAMPAH
det_table_1014/inference.pdmodel


+ 9 - 1
server.py

@@ -40,7 +40,13 @@ table_engine1 = PPStructure(layout=False,
                             use_gpu=use_gpu,
                             show_log=True,
                             det_model_dir="./det_table_1012",
-                            able_model_dir="./SLAnet_1004")
+                            table_model_dir="./SLAnet_1004")
+
+table_engine2 = PPStructure(layout=False,
+                            table=True,
+                            use_gpu=use_gpu,
+                            show_log=True,
+                            table_model_dir="./SLAnet_1011")
 
 
 class TableInfo(BaseModel):
@@ -61,6 +67,8 @@ def table(image: TableInfo):
         res = table_engine(img)
     elif image.det == 'yes':
         res = table_engine1(img)
+    elif image.det == 'spe':
+        res = table_engine2(img)
     return res[0]['res']