- 直接迁移torchvision中部分模型,resnet,vgg,alexnet等;
- 使用paddlepadle动态图模式,直接将torchvision中的resnet,vgg, alexnet代码转换为paddlepaddle代码;
- 将torchvison的resnet,vgg, alexnet模型的预训练模型参数保存为paddlepaddle动态图可读格式;
- 可以使用convert_params.py直接将pytorch模型数据转换为对应的paddlepaddle动态图数据保存;
- 修改模型保存路径,可以直接使用对应的dygraph模型即可使用加载预训练数据;
from models import resnet34 model = resnet34(pretrained=True, root="") # root 对应模型数据的保存路径
-
alexnet: 百度网盘提取码:v82t
-
vgg系列:
- vgg11: 百度网盘提取码:0db9
- vgg13: 百度网盘提取码:8xhj
- vgg16: 百度网盘提取码:5avl
- vgg19: 百度网盘提取码:rgez
-
resnet系列:
- resnet34: 百度网盘提取码:ku58
- resnet50: 百度网盘提取码:3t4s
- resnet101: 百度网盘提取码:sg8z
- resnet152: 百度网盘提取码:0r0p
-
resnest系列:
- resnest50: 百度网盘提取码:0ubv
- resnest101: 百度网盘提取码:42yr
- resnest200: 百度网盘提取码:r2cm
- resnest269: 百度网盘提取码:f2fg
-
res2net系列:
- res2net50: 百度网盘提取码:qk3g
- res2net101: 百度网盘提取码:ez0c
- res2next: 百度网盘提取码:my0j