ucalyptus / BS-Nets-Implementation-Pytorch

BS-Nets: An End-to-End Framework For Band Selection of Hyperspectral Image
MIT License
23 stars 4 forks source link

Band selection rules #7

Open ywj55555 opened 3 years ago

ywj55555 commented 3 years ago

Is your feature request related to a problem? Please describe. A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]

Describe the solution you'd like A clear and concise description of what you want to happen.

Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.

Additional context

First of all, thank you for the code. However, I found that the information entropy of each channel of the model output tensor was used when selecting the band in your code. In the original paper, “According to the learned sparse band weights, we can determine the informative bands by averaging the band weights for all the training samples. Those bands that have larger average weights are considered to be significant since they make more contributions to the reconstruction.”That means the weight of each channel output by the channel attention module is used ,this is different from the method you use,is there any reason for you to do this,please?

github-actions[bot] commented 3 years ago

Hi! thanks for your contribution!, great first issue!