glenn-jocher commited on
Commit
fdc2239
Β·
unverified Β·
1 Parent(s): 6306091

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='data/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",
 
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",