Explorar el Código

更新: 版面检测模型

zeke-chin hace 2 años
padre
commit
5059ee4b67
Se han modificado 2 ficheros con 2 adiciones y 2 borrados
  1. 1 1
      Dockerfile
  2. 1 1
      server.py

+ 1 - 1
Dockerfile

@@ -102,7 +102,7 @@ RUN git clone https://gitee.com/monkeycc/yolov5.git
 # RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/yili.pt
 # RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/best.pt
 # RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/best2.pt
-RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/yili2.pt
+RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/8-14_layout_model.pt
 
 Add . /workspace
 EXPOSE 8080

+ 1 - 1
server.py

@@ -17,7 +17,7 @@ YOLO_DIR = '/workspace/yolov5'
 # WEIGHTS = '/workspace/best.pt'
 # WEIGHTS = '/workspace/yili.pt'
 # WEIGHTS = '/workspace/best2.pt'
-WEIGHTS = '/workspace/yili2.pt'
+WEIGHTS = '/workspace/8-14_layout_model.pt'
 
 logger = logging.getLogger('log')
 logger.setLevel(logging.DEBUG)