pytorch

resnet默认输入图像通道为3通道，预训练模型也是3通道的。如果希望输入是单通道灰度图该如何修改呢？ 比较简单，只需要修改第一个卷积的输入通道即可。查看resnet的函数初始化可以看到最开始的第一个