search

icoz69 / CaNet

The code for paper "CANet: Class-Agnostic Segmentation Networks with Iterative Refinement and Attentive Few-Shot Learning"
191 stars 33 forks source link

Cannot find the attention module and code for 5-shot semantic segmentation? #17

Open Balabala-Hong opened 4 years ago

Balabala-Hong commented 4 years ago

Hi, thank you for share your awesome work. But I cannot find the attention module and code for 5-shot semantic segmentation, can you tell me where is is?

ericlearning commented 4 years ago

I also have the same problem. Any updates?

lxtGH commented 4 years ago

I aslo have the same problem. Where is the atttention module for fusion few shot results? @icoz69

icoz69 commented 4 years ago

hi. i only upload the one-shot model here. as to the k-shot loader and network, i will clean the codes and upload soon. before that you easily implement them based on the current code, such that repeating sampling support images in the loader script and adding the attention module in the network script.

yswang1717 commented 4 years ago

Could you please share the 5-shot loader and network please?

before that you easily implement them based on the current code, such that repeating sampling support images in the loader script and adding the attention module in the network script.

Can I just repeat support branch 5 times for 5-shot segmentation? Then, would the dimension of the output from the DCM module be calculated by /2(1 support, 1 query)*(5 support + 1 query) times?