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

Training from scratch #308

Open longxvu opened 3 years ago

longxvu commented 3 years ago

I tried to overfit the model by training it with 1 image (also custom class) to see if the model works. However, the detection result is bad compared to using pretrained weight. Note that my input size is different (vertically oriented), is this a reason why the model doesn't work? Also when training, the train and test loss on the same image is very different (test loss is about 15x higher than train loss) even though the functions to compute loss are the same. Anyone has the same issue?

tranleanh commented 3 years ago

I did the same thing and got the same problem with you, bro! This source may only work for converting and testing from darknet weights.