zhiyuanyou / SAFECount

[WACV 2023] Few-shot Object Counting with Similarity-Aware Feature Enhancement
Other
124 stars 22 forks source link

Correct the dimension order of resized image #24

Closed m-zheng closed 1 year ago

m-zheng commented 1 year ago

The input image is of shape [1, c, h, w]. Therefore, the resized image should follow the same dimension order [h_rsz, w_rsz].

zhiyuanyou commented 1 year ago

Thanks. It has been merged.