Skip to content

image classification models(PaddlePaddle dynamic graph) which are transferring from torchvision

Notifications You must be signed in to change notification settings

wuyang556/paddlevision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 

Repository files navigation

PaddleVision:基于paddlepaddle动态图模式的图像分类模型库

介绍
  1. 直接迁移torchvision中部分模型,resnet,vgg,alexnet等;
  2. 使用paddlepadle动态图模式,直接将torchvision中的resnet,vgg, alexnet代码转换为paddlepaddle代码;
  3. 将torchvison的resnet,vgg, alexnet模型的预训练模型参数保存为paddlepaddle动态图可读格式;
使用说明
  1. 可以使用convert_params.py直接将pytorch模型数据转换为对应的paddlepaddle动态图数据保存;
  2. 修改模型保存路径,可以直接使用对应的dygraph模型即可使用加载预训练数据;
    from models import resnet34
    model = resnet34(pretrained=True, root="")  # root 对应模型数据的保存路径
    
    
已转换的paddlepaddle动态图版预训练模型

About

image classification models(PaddlePaddle dynamic graph) which are transferring from torchvision

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages