mlzxy / devit

CoRL 2024
https://mlzxy.github.io/devit
MIT License
351 stars 46 forks source link

Inference results appear weird when there is only one category in prototype #56

Open YFZh opened 6 months ago

YFZh commented 6 months ago

I referred to that issue and made modifications to the code. I found that the prediction effect is not good, and the predicted box is often a box that is very close to the size of the image.

The following screenshot is my added code in detectron2/modeling/meta_arch/devit.py:

截屏2024-05-30 下午3 10 18

@mlzxy

I hope to receive your help. Thanks.