darthjeio t1_iywoo5p wrote on December 4, 2022 at 7:23 PM

I’m working on images (let’s say object detection) but the information is somewhat sparse (let’s say detect a white line on a noisy dark background). What would be a good model for this task in order to save computational time/resources? CNN-based SOTA models seems a bit overkill even tho I’m sure they would work. Was thinking about masking or transformers.. any idea?

zenmandala t1_iyy3jr5 wrote on December 5, 2022 at 1:10 AM

I've had success with Squeezenet for finding the origin of white circles in extremely noisy images, so maybe you could use that. Just change the last convolution in the classifier to match the desired dimensions of your output.

I was able to CPU train a solution that way. It's actually my go to for tasks like that because it seems to just do better than some larger newer networks at that sort of thing.