Browse Source

chore: 更新表格结构预测模型版本

jingze_cheng 8 months ago
parent
commit
be92760c62
2 changed files with 2 additions and 1 deletions
  1. 1 0
      Dockerfile
  2. 1 1
      server.py

+ 1 - 0
Dockerfile

@@ -100,6 +100,7 @@ stdout_logfile_maxbytes=50MB\n\
 " > /etc/supervisor/conf.d/be.conf
 " > /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-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 / && wget ftp://192.168.199.31/table-paddleocr.tgz && tar xvf table-paddleocr.tgz
+RUN mkdir -p models/table && cd models/table && wget ftp://192.168.199.31/SLANet_ch/20240222/SLANet_ch.zip && unzip SLANet_ch.zip && mv SLANet_ch SLANet_latest && rm SLANet_ch.zip
 
 
 Add . /workspace
 Add . /workspace
 EXPOSE 8080
 EXPOSE 8080

+ 1 - 1
server.py

@@ -33,7 +33,7 @@ table_engine = CustomPPStructure(layout=False,
                            use_gpu=True,
                            use_gpu=True,
                            show_log=True,
                            show_log=True,
                            use_angle_cls=True,
                            use_angle_cls=True,
-                           table_model_dir="models/table/SLANet_911")
+                           table_model_dir="models/table/SLANet_latest")
 
 
 cls_lock = threading.Lock()
 cls_lock = threading.Lock()