소스 검색

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

jingze_cheng 8 달 전
부모
커밋
bfebeb5b50
2개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      Dockerfile
  2. 1 0
      server.py

+ 1 - 0
Dockerfile

@@ -100,6 +100,7 @@ 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 mkdir -p models/det && cd models/det && wget ftp://192.168.199.31/det_pse/table/20240223/det_pse.zip && unzip det_pse.zip -d tmp && mv tmp/det_pse det_pse_quad && rm -r tmp && rm det_pse.zip
 
 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()