Update README.md
Browse files
README.md
CHANGED
|
@@ -163,6 +163,13 @@ python benchmarks.py --size 416 --model yolov4 --weights ./data/yolov4.weights
|
|
| 163 |
| YoloV4 FPS | | | |
|
| 164 |
|
| 165 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 166 |
# Transfer learning:
|
| 167 |
python train.py --weights ./data/yolov4.weights
|
| 168 |
```
|
|
|
|
| 163 |
| YoloV4 FPS | | | |
|
| 164 |
|
| 165 |
|
| 166 |
+
### Traning your own model
|
| 167 |
+
```bash
|
| 168 |
+
# Prepare your dataset
|
| 169 |
+
# If you want to train from scratch:
|
| 170 |
+
In config.py set FISRT_STAGE_EPOCHS=0
|
| 171 |
+
# Run script:
|
| 172 |
+
python train.py
|
| 173 |
# Transfer learning:
|
| 174 |
python train.py --weights ./data/yolov4.weights
|
| 175 |
```
|