![](https://img-blog.csdnimg.cn/20200303151219322.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
PyTorch基础
pytorch基础篇
玖零猴
嵌入式·ROS·CV·ML·DL·医学影像分割
展开
-
pytorch加载预训练模型
转载 2020-05-12 10:08:13 · 633 阅读 · 0 评论 -
[函数]nn.AdaptiveAvgPool2d
class torch.nn.AdaptiveAvgPool2d(output_size) 对由多个输入平面组成的输入信号应用二维自适应平均池 (2D adaptive average pooling )。 对于任何输入大小,输出的大小都是。 输出特征图的数量等于输入特征图的数量。 ...原创 2020-04-05 10:33:56 · 685 阅读 · 0 评论 -
[函数]nn.Conv2d
nn.Conv2d(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)) 参数: in_channel: 输入数据的通道数,例RGB图片通道数为3; out_channel: 输出数据的通道数,这个根据模型调整; kennel_siz...原创 2020-04-05 09:39:59 · 417 阅读 · 0 评论