-

Dockerfile

@@ -99,8 +99,8 @@ environment=CUDA_VISIBLE_DEVICES=0,PYTHONPATH=/worksapce/yolov5\n\
 # envs/py38/
 
 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/yili.pt
+RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/best.pt
 
 Add . /workspace
 EXPOSE 8080

server.py

@@ -14,8 +14,8 @@ import logging
 
 YOLO_DIR = '/workspace/yolov5'
 # WEIGHTS = '/data/yolov5/runs/train/yolov5x_layout_reuslt37/weights/best.pt'
-# WEIGHTS = '/workspace/best.pt'
-WEIGHTS = '/workspace/yili.pt'
+WEIGHTS = '/workspace/best.pt'
+# WEIGHTS = '/workspace/yili.pt'
 
 logger = logging.getLogger('log')
 logger.setLevel(logging.DEBUG)