Pytorch 加载预训练模型之ResNet系列

Pytorch 加载预训练模型之ResNet系列

参考网址：https://github.com/Cadene/pretrained-models.pytorch

1.安装方法

使用pip安装：

安装之后在python 文件中就可以使用很多的预训练模型，加载预训练模型的好处：加速模型收敛；使模型得到更好的局部解；引入相关领域的知识。

使用的思路

数据预处理；修改类别数

数据预处理不是这里的重点，这里不做过多的介绍，下面简单引用网址提供的方法

这里稍作修改：

Note: 从网络得到输出是（1，num_class）的大小，根据自己实际数据集的大小进行调整和训练；

除了resnet-50,还有其他相关的模型，使用的方法类似

第30,31行中，提供了提取feature map的函数和获得logit的函数