search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.07k stars 180 forks source link

Multiple classes on one object detected #78

Closed Testbild closed 2 years ago

Testbild commented 2 years ago

Hello,

I am currently facing the issue, that the model is drawing multiple boxes around the same image, even though the object is not in the picture, something like this, where e.g.:

Blue = Cat Red = Dog

Untitled

But both are on the cat. Do you have an idea how to prevent this from happening? Is it maybe possible to give a minimum threshold of confidence for boxes to be considered and drawn?

Thanks and best regards

yangxue0827 commented 2 years ago

This shows that the classification of the detector is not good enough to distinguish these classes. In your example, the classification of cats and dogs is a challenging task.