此博文是修改https://blog.csdn.net/jiacong_wang/article/details/105631229
这位大大的博文而成的,自己根据自己的情况稍微加了点东西
要修改的地方有4处
1.修改网络第一层,把3通道改为1
法一:直接在定义网络的地方修改
self.conv1 = nn.Conv2d(1, self.in_channel, kernel_size=7, stride=2,padding=3, bias=False)
法二:在调用网络模型的地方修改
model = resnet50()
model.conv1 = nn.Conv2d(1, 64, kernel_size=7