fix； 1

detect.py

@@ -246,7 +246,7 @@ def parse_opt():
     # 定义命令行可以传入的参数
     parser = argparse.ArgumentParser()
     # 权重
-    parser.add_argument('--weights', nargs='+', type=str, default='runs/train/mask_new_300/weights/best.pt',
+    parser.add_argument('--weights', nargs='+', type=str, default='runs/train/exp4/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')

runs/train/mask_300/hyp.yaml

-lr0: 0.01
-lrf: 0.01
-momentum: 0.937
-weight_decay: 0.0005
-warmup_epochs: 3.0
-warmup_momentum: 0.8
-warmup_bias_lr: 0.1
-box: 0.05
-cls: 0.5
-cls_pw: 1.0
-obj: 1.0
-obj_pw: 1.0
-iou_t: 0.2
-anchor_t: 4.0
-fl_gamma: 0.0
-hsv_h: 0.015
-hsv_s: 0.7
-hsv_v: 0.4
-degrees: 0.0
-translate: 0.1
-scale: 0.5
-shear: 0.0
-perspective: 0.0
-flipud: 0.0
-fliplr: 0.5
-mosaic: 1.0
-mixup: 0.0
-copy_paste: 0.0

runs/train/mask_300/opt.yaml

-weights: yolov5s.pt
-cfg: owndata/yolov5s.yaml
-data: owndata/data.yaml
-hyp:
-  lr0: 0.01
-  lrf: 0.01
-  momentum: 0.937
-  weight_decay: 0.0005
-  warmup_epochs: 3.0
-  warmup_momentum: 0.8
-  warmup_bias_lr: 0.1
-  box: 0.05
-  cls: 0.5
-  cls_pw: 1.0
-  obj: 1.0
-  obj_pw: 1.0
-  iou_t: 0.2
-  anchor_t: 4.0
-  fl_gamma: 0.0
-  hsv_h: 0.015
-  hsv_s: 0.7
-  hsv_v: 0.4
-  degrees: 0.0
-  translate: 0.1
-  scale: 0.5
-  shear: 0.0
-  perspective: 0.0
-  flipud: 0.0
-  fliplr: 0.5
-  mosaic: 1.0
-  mixup: 0.0
-  copy_paste: 0.0
-epochs: 300
-batch_size: -1
-imgsz: 640
-rect: false
-resume: false
-nosave: false
-noval: false
-noautoanchor: false
-noplots: false
-evolve: null
-bucket: ''
-cache: null
-image_weights: false
-device: ''
-multi_scale: false
-single_cls: false
-optimizer: SGD
-sync_bn: false
-workers: 0
-project: runs\train
-name: exp
-exist_ok: false
-quad: false
-cos_lr: false
-label_smoothing: 0.0
-patience: 100
-freeze:
-- 0
-save_period: -1
-seed: 0
-local_rank: -1
-entity: null
-upload_dataset: false
-bbox_interval: -1
-artifact_alias: latest
-save_dir: runs\train\exp3

runs/train/mask_new_150/hyp.yaml

-lr0: 0.01
-lrf: 0.01
-momentum: 0.937
-weight_decay: 0.0005
-warmup_epochs: 3.0
-warmup_momentum: 0.8
-warmup_bias_lr: 0.1
-box: 0.05
-cls: 0.5
-cls_pw: 1.0
-obj: 1.0
-obj_pw: 1.0
-iou_t: 0.2
-anchor_t: 4.0
-fl_gamma: 0.0
-hsv_h: 0.015
-hsv_s: 0.7
-hsv_v: 0.4
-degrees: 0.0
-translate: 0.1
-scale: 0.5
-shear: 0.0
-perspective: 0.0
-flipud: 0.0
-fliplr: 0.5
-mosaic: 1.0
-mixup: 0.0
-copy_paste: 0.0

runs/train/mask_new_300/hyp.yaml

-lr0: 0.01
-lrf: 0.01
-momentum: 0.937
-weight_decay: 0.0005
-warmup_epochs: 3.0
-warmup_momentum: 0.8
-warmup_bias_lr: 0.1
-box: 0.05
-cls: 0.5
-cls_pw: 1.0
-obj: 1.0
-obj_pw: 1.0
-iou_t: 0.2
-anchor_t: 4.0
-fl_gamma: 0.0
-hsv_h: 0.015
-hsv_s: 0.7
-hsv_v: 0.4
-degrees: 0.0
-translate: 0.1
-scale: 0.5
-shear: 0.0
-perspective: 0.0
-flipud: 0.0
-fliplr: 0.5
-mosaic: 1.0
-mixup: 0.0
-copy_paste: 0.0

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=300, help='total training epochs')
+    parser.add_argument('--epochs', type=int, default=100, 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')