search

zhiyuanyou / SAFECount

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

Hello , I coudn't found the Similarity Comparison Module (SCM) and the Feature Enhancement Module (FEM) in your code. #21

Closed Lay-best closed 1 year ago

Lay-best commented 1 year ago

Could you please tell me where are these module in your code?

zhiyuanyou commented 1 year ago

calculate similarity (attention): https://github.com/zhiyuanyou/SAFECount/blob/main/models/safecount.py#L195 score normalization: https://github.com/zhiyuanyou/SAFECount/blob/main/models/safecount.py#L225 similarity weighted aggregation: https://github.com/zhiyuanyou/SAFECount/blob/main/models/safecount.py#L236 fuse feature: https://github.com/zhiyuanyou/SAFECount/blob/main/models/safecount.py#L153

Lay-best commented 1 year ago

wow, I got it , really appreciate your help!