Pytorch 加载预训练模型之ResNet系列
参考网址:https://github.com/Cadene/pretrained-models.pytorch
1.安装方法
使用pip安装:
安装之后在python 文件中就可以使用很多的预训练模型,加载预训练模型的好处:加速模型收敛;使模型得到更好的局部解;引入相关领域的知识。
使用的思路
数据预处理;修改类别数
数据预处理不是这里的重点,这里不做过多的介绍,下面简单引用网址提供的方法
这里稍作修改:
Note: 从网络得到输出是(1,num_class)的大小,根据自己实际数据集的大小进行调整和训练;
除了resnet-50,还有其他相关的模型,使用的方法类似
第30,31行中,提供了提取feature map的函数和获得logit的函数
领取专属 10元无门槛券
私享最新 技术干货