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