search

ShichengChen / WaveUNet

implement Wave-U-Net by pytorch
MIT License
56 stars 9 forks source link

What is the SELayer? The improved result. #1

Open ronggong opened 5 years ago

ronggong commented 5 years ago

Hi Shicheng,

Thanks for the implementation of WaveUNet. I have a problem about the SELayer https://github.com/ShichengChen/WaveUNet/blob/54df9d63378ff5db251f0a793028d30608efa64e/modelStruct/utils.py#L63

Could you give a little bit detail of this layer? Why you used it?

Another question is that your improved result is based on which model structure?

Thanks!

ShichengChen commented 5 years ago

Hi, Rong gong,

Thank you for your interest. SELayer is from the paper https://arxiv.org/pdf/1709.01507.pdf It increases the accuracy of image classification task, so I want to have a try on my work. I remembered that it is from unet.py or resunet.py