search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.24k stars 1.24k forks source link

Please update code for tiny-yolo training! #199

Open panguxiaoshen opened 4 years ago

panguxiaoshen commented 4 years ago

In dataset.py, the code is only given for 3 featuremap in 83~183 lines, please update it. @hunglc007

czy9987 commented 4 years ago

Excuse me, have you implemented the training for the tiny version?

hhk7734 commented 4 years ago

@panguxiaoshen @czy9987

If you are interested in tiny training, https://wiki.loliot.net/docs/etc/project/yolov4/yolov4-training

In guide, Replacing yolo.make_model(activation1="relu") with yolo.make_model() results in yolov4-tiny.