RuntimeError: Given groups=1, weight of size [64, 1, 9, 9, 7], expected input维度错误问题

项目场景：

复现网络代码3D-FCNN网络模型，采用数据集为Houston,shape 为Channel-band-h-w （1-48-36-36）,网络模型结构采用64-32-9-1。

---

问题描述

错误：RuntimeError: Given groups=1, weight of size [64, 1, 9, 9, 7], expected input[1, 28, 48, 36, 36] to have 1 channels, but got 28 channels instead

---

原因分析：

网络输入的维度是一个通道，但是实际输入的确实28，这里的28我设置的批量大小，所以输出缺少一个通道维度，因为高光谱数据集一般是通道数波段数长*宽，通道数为1，往往会省略。需要自己加入

---

解决方案：

在数据维度加入一个通道维度，维度数为1。这是其中一个解决方案，第二种在网络里面增加一个通道维（未实现）

train_data, train_label = read_training_data(
    r"C:\Users\Duanyikai\Desktop\traindata.h5")
print(type(train_data),type(train_label),train_data.shape, train_label.shape)
train_data = numpy.expand_dims(train_data,1) #扩展一个通道维
train_label = numpy.expand_dims(train_label, 1)
print('改完之后的维度',train_data.shape, train_label.shape)

output：
更改之前的维度 <class 'numpy.ndarray'> <class 'numpy.ndarray'> (2550, 48, 36, 36) (2550, 48, 36, 36)
改完之后的维度 <class 'numpy.ndarray'> <class 'numpy.ndarray'> (2550, 1, 48, 36, 36) (2550, 1, 48, 36, 36)

问题解决