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

[Balabala Hong]

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?

[Ku Seung Mo]

I also have the same problem. Any updates?

[Xiangtai Li]

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

[icoz69]

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.

[Yooseung Wang]

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?