Update README.md
Browse files
README.md
CHANGED
@@ -162,18 +162,12 @@ python benchmarks.py --size 416 --model yolov4 --weights ./data/yolov4.weights
|
|
162 |
| YoloV4 FPS | | | |
|
163 |
|
164 |
|
165 |
-
### Traning your own model
|
166 |
-
```bash
|
167 |
-
# Prepare your dataset
|
168 |
-
# If you want to train from scratch:
|
169 |
-
In config.py set FISRT_STAGE_EPOCHS=0
|
170 |
-
# Run script:
|
171 |
-
python train.py
|
172 |
# Transfer learning:
|
173 |
python train.py --weights ./data/yolov4.weights
|
174 |
```
|
175 |
The training performance is not fully reproduced yet, so I recommended to use Alex's [Darknet](https://github.com/AlexeyAB/darknet) to train your own data, then convert the .weights to tensorflow or tflite.
|
176 |
|
|
|
177 |
### References
|
178 |
|
179 |
* YOLOv4: Optimal Speed and Accuracy of Object Detection [YOLOv4](https://arxiv.org/abs/2004.10934).
|
|
|
162 |
| YoloV4 FPS | | | |
|
163 |
|
164 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
165 |
# Transfer learning:
|
166 |
python train.py --weights ./data/yolov4.weights
|
167 |
```
|
168 |
The training performance is not fully reproduced yet, so I recommended to use Alex's [Darknet](https://github.com/AlexeyAB/darknet) to train your own data, then convert the .weights to tensorflow or tflite.
|
169 |
|
170 |
+
|
171 |
### References
|
172 |
|
173 |
* YOLOv4: Optimal Speed and Accuracy of Object Detection [YOLOv4](https://arxiv.org/abs/2004.10934).
|