SamMorgan commited on
Commit
b9d4c0e
·
1 Parent(s): 0904bcf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -7
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).