zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.21k stars 1.27k forks source link

[Help Wanted]Problem while training on customize datasaet #632

Open laplaceson opened 3 years ago

laplaceson commented 3 years ago

Hi @zylo117 , First thanks for your awesome work! I try to use pretrained weight D0 model on my custom dataset. However, it always gives reg loss 0 . So i try to use other method to modify the anchors_scales and anchors_ratios. (Since my object are small and sparse). The reg loss goes non-zero and converge. But i figure out my validation reg loss and training reg loss have big difference. image And finally i use debug mode (Or using weight to efficientdet_test.py ) to visualize it. But get nothing detected. Do you have any suggestion? Thanks By the way, my task is single class object detection(object size often not big).

laplaceson commented 3 years ago

Another question, i see some of your comment says it's hard to detect smaller 32 32 , is this means that under 512 512 's image, if the annotation is smaller than 32 * 32, it's hard to detect in this network? Thanks

zylo117 commented 3 years ago

32x32 under 512x512