SamMorgan commited on
Commit
2db59c9
·
1 Parent(s): 2d2c489

Update README.md

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