fixed: 重新提交口罩数据集

.gitignore

@@ -255,3 +255,4 @@
 # crashlytics.properties
 # crashlytics-build.properties
 # fabric.properties
+/owndata/mask/

detect.py

@@ -246,7 +246,7 @@ def parse_opt():
     # 定义命令行可以传入的参数
     parser = argparse.ArgumentParser()
     # 权重
-    parser.add_argument('--weights', nargs='+', type=str, default='runs/train/mask_new_150/weights/best.pt',
+    parser.add_argument('--weights', nargs='+', type=str, default='runs/train/mask_new_300/weights/best.pt',
                         help='model path or triton URL')
     parser.add_argument('--source', type=str, default='owndata/test/video', help='file/dir/URL/glob/screen/0(webcam)')
     parser.add_argument('--data', type=str, default=ROOT / 'data/coco128.yaml', help='(optional) dataset.yaml path')

train.py

@@ -464,7 +464,7 @@ def parse_opt(known=False):
     parser.add_argument('--cfg', type=str, default='owndata/yolov5s.yaml', help='model.yaml path')
     parser.add_argument('--data', type=str, default='owndata/mask.yaml', help='dataset.yaml path')
     parser.add_argument('--hyp', type=str, default=ROOT / 'data/hyps/hyp.scratch-low.yaml', help='hyperparameters path')
-    parser.add_argument('--epochs', type=int, default=100, help='total training epochs')
+    parser.add_argument('--epochs', type=int, default=300, help='total training epochs')
     parser.add_argument('--batch-size', type=int, default=-1, help='total batch size for all GPUs, -1 for autobatch')
     parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=640, help='train, val image size (pixels)')
     parser.add_argument('--rect', action='store_true', help='rectangular training')