Dockerfile

@@ -93,7 +93,7 @@ startretries=0\n\
 redirect_stderr=true\n\
 stdout_logfile=/var/log/be.log\n\
 stdout_logfile_maxbytes=50MB\n\
-environment=CUDA_VISIBLE_DEVICES=0,PYTHONUNBUFFERED=1,PYTHONPATH=/worksapce/yolov5\n\
+environment=CUDA_VISIBLE_DEVICES=0,PYTHONPATH=/worksapce/yolov5\n\
 " > /etc/supervisor/conf.d/be.conf
 
 # envs/py38/
@@ -102,9 +102,9 @@ 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/8-14_layout_model.pt
+RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/yili2.pt
 
-ADD . /workspace
+Add . /workspace
 EXPOSE 8080
 
 

server.py

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