PyTorch教程:21个项目玩转PyTorch实战
上QQ阅读APP看书,第一时间看更新

2.1.3 定义卷积神经网络

复制1.2.1节神经网络模块相关的代码,并修改它为三通道的图片(在此之前它被定义为单通道),代码如下:

将nn.Conv2d的第一个参数从1改成3,这个卷积模块就支持输入三通道的图片,之后就和第1章所使用的模型结构一样了。需要注意的是,这里展示了pooling的另外一种使用方法,把Max Pooling操作定义成一个网络模块,而不像第1章那样调用函数完成pooling操作。但两者的实现效果是一样的。