facebookresearch / detr

End-to-End Object Detection with Transformers
Apache License 2.0
13.09k stars 2.37k forks source link

Why YOLOS model with balanced Dataset Predicting single class only #542

Open GazalaSayyad opened 1 year ago

GazalaSayyad commented 1 year ago

How to model predicts only one class

Fine Tuned YOLOS model with 120 images in training. 80 images in validation. In training 60 of class 1 and half are class 2 labeld.. same way for validation. But after model trained at prediction it predicting all 80 images with class 1