Same no of training labels predicted

XinzeLee / RotateObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.

113 stars 22 forks source link

Same no of training labels predicted #24

Open satpalsr opened 2 years ago

satpalsr commented 2 years ago

Hey @XinzeLee, I followed Rotate-tutorial 1 colab to train the model for 100 epochs but I observe the same low number of labels predicted at each epoch:

I have also tried it for my custom dataset & a similar problem persists. What are your thoughts on it?