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 |
```
|