首页 > TAG信息列表 > HybridSN
HybridSN代码修改
研究小组里初学习深度学习的同学我都布置写过 HybridSN 的代码:https://github.com/OUCTheoryGroup/colab_demo/blob/master/202003_models/HybridSN_GRSL2020.ipynb 最近做SRDP的同学反映,跑 Pavia 数据集的时候内存会爆,主要原因是 createImageCubes 这个函数有个地方: patchesData《HybridSN: Exploring 3-D–2-DCNN Feature Hierarchy for Hyperspectral Image Classification》论文学习笔记
文章目录 论文基本信息研究背景读完摘要后的疑问读完文章后对上述问题的回答高光谱基础知识HybridSN 实验验证数据集描述分类结果 代码分析存在的疑惑 论文基本信息 标题:HybridSN: Exploring 3-D–2-DCNN Feature Hierarchy for Hyperspectral Image Classification作者:S完善HybridSN 高光谱分类网络
代码 ! wget http://www.ehu.eus/ccwintco/uploads/6/67/Indian_pines_corrected.mat ! wget http://www.ehu.eus/ccwintco/uploads/c/c4/Indian_pines_gt.mat ! pip install spectral import numpy as np import matplotlib.pyplot as plt import scipy.io as sio from sklearHybridSN+2D SENet
1、在HybridSN模块中添加注意力机制: class_num = 16 rate = 16 # SENet论文中建议rate=sqrt(256)=16 class HybridSN(nn.Module): def __init__(self): super(HybridSN, self).__init__() self.conv1 = nn.Conv3d(1, 8, kernel_size=(7, 3, 3), stride=1, padding=0)HybridSN尝试加入SENet与dropout的一些坑
尝试在HybridSN 高光谱分类网络卷积层后加入SENet模块,代码如下: class SELayer(nn.Module): def __init__(self, channel, reduction=16): super(SELayer, self).__init__() self.avg_pool = nn.AdaptiveAvgPool2d(1) self.fc = nn.Sequential(