Create `data/hyps` directory (#3747)
Browse files
.gitignore
CHANGED
@@ -19,26 +19,18 @@
|
|
19 |
*.avi
|
20 |
*.data
|
21 |
*.json
|
22 |
-
|
23 |
*.cfg
|
24 |
!cfg/yolov3*.cfg
|
25 |
|
26 |
storage.googleapis.com
|
27 |
runs/*
|
28 |
data/*
|
|
|
29 |
!data/images/zidane.jpg
|
30 |
!data/images/bus.jpg
|
31 |
-
!data/coco.names
|
32 |
-
!data/coco_paper.names
|
33 |
-
!data/coco.data
|
34 |
-
!data/coco_*.data
|
35 |
-
!data/coco_*.txt
|
36 |
-
!data/trainvalno5k.shapes
|
37 |
!data/*.sh
|
38 |
|
39 |
-
pycocotools/*
|
40 |
results*.txt
|
41 |
-
gcp_test*.sh
|
42 |
|
43 |
# Datasets -------------------------------------------------------------------------------------------------------------
|
44 |
coco/
|
|
|
19 |
*.avi
|
20 |
*.data
|
21 |
*.json
|
|
|
22 |
*.cfg
|
23 |
!cfg/yolov3*.cfg
|
24 |
|
25 |
storage.googleapis.com
|
26 |
runs/*
|
27 |
data/*
|
28 |
+
!data/hyps/*
|
29 |
!data/images/zidane.jpg
|
30 |
!data/images/bus.jpg
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
!data/*.sh
|
32 |
|
|
|
33 |
results*.txt
|
|
|
34 |
|
35 |
# Datasets -------------------------------------------------------------------------------------------------------------
|
36 |
coco/
|
data/{hyp.finetune.yaml β hyps/hyp.finetune.yaml}
RENAMED
File without changes
|
data/{hyp.finetune_objects365.yaml β hyps/hyp.finetune_objects365.yaml}
RENAMED
File without changes
|
data/{hyp.scratch.yaml β hyps/hyp.scratch.yaml}
RENAMED
File without changes
|
train.py
CHANGED
@@ -484,7 +484,7 @@ def parse_opt(known=False):
|
|
484 |
parser.add_argument('--weights', type=str, default='yolov5s.pt', help='initial weights path')
|
485 |
parser.add_argument('--cfg', type=str, default='', help='model.yaml path')
|
486 |
parser.add_argument('--data', type=str, default='data/coco128.yaml', help='dataset.yaml path')
|
487 |
-
parser.add_argument('--hyp', type=str, default='data/hyp.scratch.yaml', help='hyperparameters path')
|
488 |
parser.add_argument('--epochs', type=int, default=300)
|
489 |
parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs')
|
490 |
parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='[train, test] image sizes')
|
|
|
484 |
parser.add_argument('--weights', type=str, default='yolov5s.pt', help='initial weights path')
|
485 |
parser.add_argument('--cfg', type=str, default='', help='model.yaml path')
|
486 |
parser.add_argument('--data', type=str, default='data/coco128.yaml', help='dataset.yaml path')
|
487 |
+
parser.add_argument('--hyp', type=str, default='data/hyps/hyp.scratch.yaml', help='hyperparameters path')
|
488 |
parser.add_argument('--epochs', type=int, default=300)
|
489 |
parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs')
|
490 |
parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='[train, test] image sizes')
|
tutorial.ipynb
CHANGED
@@ -932,7 +932,7 @@
|
|
932 |
"\u001b[34m\u001b[1mgithub: \u001b[0mup to date with https://github.com/ultralytics/yolov5 β
\n",
|
933 |
"YOLOv5 π v5.0-158-g78cf488 torch 1.8.1+cu101 CUDA:0 (Tesla V100-SXM2-16GB, 16160.5MB)\n",
|
934 |
"\n",
|
935 |
-
"Namespace(adam=False, artifact_alias='latest', batch_size=16, bbox_interval=-1, bucket='', cache_images=True, cfg='', data='./data/coco128.yaml', device='', entity=None, epochs=3, evolve=False, exist_ok=False, global_rank=-1, hyp='
|
936 |
"\u001b[34m\u001b[1mtensorboard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n",
|
937 |
"2021-06-08 17:00:55.016221: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0\n",
|
938 |
"\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.2, 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\n",
|
|
|
932 |
"\u001b[34m\u001b[1mgithub: \u001b[0mup to date with https://github.com/ultralytics/yolov5 β
\n",
|
933 |
"YOLOv5 π v5.0-158-g78cf488 torch 1.8.1+cu101 CUDA:0 (Tesla V100-SXM2-16GB, 16160.5MB)\n",
|
934 |
"\n",
|
935 |
+
"Namespace(adam=False, artifact_alias='latest', batch_size=16, bbox_interval=-1, bucket='', cache_images=True, cfg='', data='./data/coco128.yaml', device='', entity=None, epochs=3, evolve=False, exist_ok=False, global_rank=-1, hyp='hyp.scratch.yaml', image_weights=False, img_size=[640, 640], label_smoothing=0.0, linear_lr=False, local_rank=-1, multi_scale=False, name='exp', noautoanchor=False, nosave=False, notest=False, project='runs/train', quad=False, rect=False, resume=False, save_dir='runs/train/exp', save_period=-1, single_cls=False, sync_bn=False, total_batch_size=16, upload_dataset=False, weights='yolov5s.pt', workers=8, world_size=1)\n",
|
936 |
"\u001b[34m\u001b[1mtensorboard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n",
|
937 |
"2021-06-08 17:00:55.016221: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0\n",
|
938 |
"\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.2, 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\n",
|