pretrainedmodels 库使用
prtrainedmodels 是一个基于pytorch框架构建的预训练模型使用的库,比官方torchvision中的预训练模型更加丰富。
github项目地址:https://github.com/Cadene/pretrained-models.pytorch
安装
pip install pretrainedmodels
获取所有预训练模型名称
加载基于imagenet的预训练模型
或者
预测
微调模型
直接使用结构
不使用预训练模型的话,也可以直接使用这个模型的结构,根据分类数设置自己最后一层的神经元个数。