Went-Liang / UnSniffer

[CVPR 2023] Official Pytorch code for Unknown Sniffer for Object Detection: Don’t Turn a Blind Eye to Unknown Objects
https://arxiv.org/pdf/2303.13769v1.pdf
Apache License 2.0
68 stars 9 forks source link

Run inference for a single image #20

Open emmanuelol opened 8 months ago

emmanuelol commented 8 months ago

Hi, first, I must recognize that it is an excellent paper. I reached this because I've been stuck trying to perform the inference and draw the unknown objects for at least two weeks. By reviewing the repository, when you run the apply_net.py, two JSON files are created, one for the known objects and the other that includes the unknown objects. Based on that, I've been trying to run the inference of a single image and draw the bounding boxes of the known and unknown objects. The first ones are easy to get, but I need help with the unknown objects. I already explore the methods 'OOD_detection_postprocessing,' 'probabilistic_detector_postprocess,' 'mergeIDOOD,' etc. I tried to use the scores and complete scores to get the unknown, but if some unknown objects are detected, most unknown objects overlap with the known ones. What do I need to include to identify the unknown objects?

emmanuelol commented 8 months ago

issue unsniffer Here is an example on how the inference is looking 😭