Forráskód Böngészése

chore: 更新表格文本检测模型版本

jingze_cheng 9 hónapja
szülő
commit
a3d856358d
2 módosított fájl, 4 hozzáadás és 0 törlés
  1. 3 0
      Dockerfile
  2. 1 0
      server.py

+ 3 - 0
Dockerfile

@@ -100,6 +100,9 @@ stdout_logfile_maxbytes=50MB\n\
 " > /etc/supervisor/conf.d/be.conf
 RUN cd / && wget ftp://192.168.199.31/table-paddleclas.tgz && tar xvf table-paddleclas.tgz
 RUN cd / && wget ftp://192.168.199.31/table-paddleocr.tgz && tar xvf table-paddleocr.tgz
+RUN cd models/det && wget ftp://192.168.199.31/det_pse/quad/20240205/det_pse.zip && unzip det_pse.zip -d tmp && mv tmp/det_pse det_pse_quad && rm -r tmp && rm det_pse.zip
+RUN sed -i "s#os.chdir('ppocr/postprocess/pse_postprocess/pse')#os.chdir('PaddleOCR/ppocr/postprocess/pse_postprocess/pse')#g" PaddleOCR/ppocr/postprocess/pse_postprocess/pse/__init__.py; \
+    cat PaddleOCR/ppocr/postprocess/pse_postprocess/pse/__init__.py
 
 Add . /workspace
 EXPOSE 8080

+ 1 - 0
server.py

@@ -33,6 +33,7 @@ table_engine = PPStructure(layout=False,
                            use_gpu=True,
                            show_log=True,
                            use_angle_cls=True,
+                           det_model_dir="models/det/det_pse_quad",
                            table_model_dir="models/table/SLANet_911")
 
 cls_lock = threading.Lock()