▲ 0 r/computervision
Fine tuning yolo to find people in industrial environment
I am a student and I am trying to fine tune yolo to find people in my very high resolution industrial pictures. Without fine tuning, I get a lot of false positives because of tubes and pipes (and if I raise the confidence i don’t find the people). So I fine tuned yolo. The problem is that I have very few images with people (just 20 tiles with humans and I have 750 high res pictures I slice in tiles).
I used my 20 humans to train/val yolo and about 2000 tiles with nothing. When I test again on all my HR images and I have fewer false positives and almost all humans. But I guess it’s overfitting because it runs on the tiles with humans used to train yolo.
What would you do?
Thanks
u/Cold-Act1693 — 2 days ago