Commit
•
b7ac446
1
Parent(s):
a557b7d
update README.md
Browse files
README.md
CHANGED
@@ -74,9 +74,12 @@ Results saved to /content/yolov5/inference/output
|
|
74 |
|
75 |
## Reproduce Our Training
|
76 |
|
77 |
-
|
78 |
```bash
|
79 |
-
$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size
|
|
|
|
|
|
|
80 |
```
|
81 |
<img src="https://user-images.githubusercontent.com/26833433/84186698-c4d54d00-aa45-11ea-9bde-c632c1230ccd.png" width="900">
|
82 |
|
|
|
74 |
|
75 |
## Reproduce Our Training
|
76 |
|
77 |
+
Download [COCO](https://github.com/ultralytics/yolov5/blob/master/data/get_coco2017.sh), install [Apex](https://github.com/NVIDIA/apex) and run command below. Training times for yolov5s/m/l/x are 2/4/6/8 days on a single V100 (multi-GPU times faster). Use the largest `--batch-size` your GPU allows (batch sizes shown for 16 GB devices).
|
78 |
```bash
|
79 |
+
$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 64
|
80 |
+
yolov5m 48
|
81 |
+
yolov5l 32
|
82 |
+
yolov5x 16
|
83 |
```
|
84 |
<img src="https://user-images.githubusercontent.com/26833433/84186698-c4d54d00-aa45-11ea-9bde-c632c1230ccd.png" width="900">
|
85 |
|