Hello everyone,
I’m working with the Faster RCNN to predict the bounding boxes of some particular objects inside a fixed environment. Basically what I want to predict are out-of-place bolts inside a scene. I already trained my model with synthetic images and validating it on real images and fortunately I get some discreet results. The detector is quite confident on locating the bolts (scores > 0.98) but it gets confused with objects with similar shape. An example could be the rivets.
I created my dataset including only images with objects and their bounding boxes but is there a way to actually “suggest” the model even to not consider some objects instead of telling it to locate those bolts?
Thanks in advance