ppogg / YOLOv5-Lite

🍅🍅🍅YOLOv5-Lite: Evolved from yolov5 and the size of model is only 900+kb (int8) and 1.7M (fp16). Reach 15 FPS on the Raspberry Pi 4B~
GNU General Public License v3.0
2.28k stars 407 forks source link

wrong parameters in train.py #237

Open xibici opened 1 year ago

xibici commented 1 year ago

460 parser.add_argument('--cfg', type=str, default='models/v5ite-s.yaml', help='model.yaml path')

should be

460 parser.add_argument('--cfg', type=str, default='models/v5Lite-s.yaml', help='model.yaml path')