search

ZhangGongjie / Meta-DETR

[T-PAMI 2022] Meta-DETR for Few-Shot Object Detection: Official PyTorch Implementation
MIT License
388 stars 84 forks source link

Regarding Background Encoding and Prototype #63

Closed michaelku1 closed 1 year ago

michaelku1 commented 1 year ago

Hi, Thanks for your work. I could not seem to find where background prototype was in the code. I was wondering how you compute it as mentioned in the paper? Many Thanks!

ZhangGongjie commented 1 year ago

Thanks for your interest in our work!

In models/attention.py Line #44-45, we define the background prototype (learnable parameters) and task encodings for the background (fixed at zeros).

michaelku1 commented 1 year ago

Thank you!