search

Haochen-Wang409 / U2PL

[CVPR'22] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels
Apache License 2.0
426 stars 59 forks source link

Question about the shape of prototype #132

Closed Hugo-cell111 closed 1 year ago

Hugo-cell111 commented 1 year ago

Hi! In the code the prototype is initialized as torch.zeros(num_classes, num_queries, 1, 256), and I don't know why there is an extra shape of num_queries. I think the prototype should be set for each class, so it should be torch.zeros(num_classes, 1, 256), but I may make some misunderstanding. Could you explain it? Thanks a lot!

Haochen-Wang409 commented 1 year ago

Each query from the same class shares the same prototype. We simply repeat them.