【模型转换】PyTorch模型转Caffe模型

一、Caffe模型转换源码下载

GitHub:https://github.com/xxradon/PytorchToCaffe

GitHub上拉取PytorchToCaffe的源码,将Caffe文件夹和pytorch_to_caffe.py文件放到项目根目录

二、将VGG的PyTorch模型转为Caffe模型

cpp 复制代码
import sys
sys.path.insert(0,'.')
import torch
from torch.autograd import Variable
from torchvision.models.vgg import vgg11_bn
import pytorch_to_caffe

if __name__=='__main__':
    name='vgg11_bn'
    net=vgg11_bn(True)
    input=Variable(torch.ones([1,3,224,224]))
    pytorch_to_caffe.trans_net(net,input,name)
    pytorch_to_caffe.save_prototxt('{}.prototxt'.format(name))
    pytorch_to_caffe.save_caffemodel('{}.caffemodel'.format(name))
相关推荐
果冻人工智能10 分钟前
法官们终于似乎明白了:如果没有复制,那就没有版权
人工智能
tle_sammy11 分钟前
AI 重构老旧系统:创业新曙光
人工智能·重构
果冻人工智能13 分钟前
什么是 MCP,以及你为什么该关注它
人工智能
誉鏐18 分钟前
PyTorch复现逻辑回归
人工智能·pytorch·逻辑回归
正脉科工 CAE仿真21 分钟前
基于ANSYS 概率设计和APDL编程的结构可靠性设计分析
人工智能·python·算法
EasyGBS26 分钟前
视频设备轨迹回放平台EasyCVR打造视频智能融合新平台,驱动智慧机场迈向数字新时代
网络·人工智能·安全·音视频
Chaos_Wang_32 分钟前
NLP高频面试题(三十三)——Vision Transformer(ViT)模型架构介绍
人工智能·自然语言处理·transformer
新知图书44 分钟前
OpenCV单窗口显示多图片
人工智能·opencv·计算机视觉
荷包蛋蛋怪1 小时前
【北京化工大学】 神经网络与深度学习 实验6 MATAR图像分类
人工智能·深度学习·神经网络·opencv·机器学习·计算机视觉·分类
小马哥编程1 小时前
【软测】AI助力测试用例
人工智能·测试用例