|
@@ -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)
|