BangguWu / ECANet

Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks
MIT License
1.24k stars 197 forks source link

ECA-NS implementation #36

Open gksruf opened 3 years ago

gksruf commented 3 years ago

Thanks for your great works.

I want to apply ECA-NS. I found previous issue and below was your answer. Use nn.functional.unfold() and conv1d with inchannel=C, outchannel=C and group=C. I can't get it how to use it. is there any code for it?

BangguWu commented 3 years ago

Thank you for your attention, we have updated the implementation of ECA-NS in models.

whjzsy commented 3 years ago

Thank you for your good job! I am a beginer, what is the ECA-NS?

gksruf commented 3 years ago

@whjzsy One of ECA's variant. Read the paper and you will find it.

gksruf commented 3 years ago

@BangguWu I did ECA-NS using different way (but similar with yours) but it was slow compared to SE or ECA. As kernel size become larger, it would be much slower. Am i right?