search

djukicn / loca

LOCA - A Low-Shot Object Counting Network With Iterative Prototype Adaptation (ICCV 2023)
MIT License
44 stars 4 forks source link

Fix dimension inconsistency error. #6

Closed Sunhill666 closed 7 months ago

Sunhill666 commented 7 months ago

The bounding boxes missing a dimension and the logical judgment of zero_shot is wrong.

Sunhill666 commented 7 months ago

The code is worked very well on torch 2.2.0 & torchvision 0.17.0, except the commits made above.