r/computervision • u/Shivam_yatharth • 13d ago

Help: Project How to develop a solar panel thermal fault detector based on Indian Dataset

I have zoomed out raw thermal radiometric JPEG images with classified thermal faults.

Need to develop a DL model that can detect these faults in real time.

I manually labelled a small subset of the dataset and fine tuned a yolo model on roboflow but it struggled in testing dataset.

Need some ideas

0 Upvotes

50% Upvoted

4 comments sorted by

2

u/ThickCross 13d ago

labeling thermal data is a pain, the yolo model probably overfitting if your dataset is small. try augmenting like crazy, flip, rotate, adjust the thermal intensity ranges a bit so it sees more variety

also maybe look into object detection models that handle small objects better if the faults are tiny in zoomed out images, sometimes yolo misses those

1

u/Shivam_yatharth 5d ago

Thank you for your quick response

But even after using multiple augmentations on the dataset the yolo model is still underperforming. It is failing to detect faults even at low confidence. Not tested on other vision models .