【模型转换】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))
相关推荐
张较瘦_1 分钟前
[论文阅读] 人工智能 + 软件工程 | TDD痛点破解:LLM自动生成测试骨架靠谱吗?静态分析+专家评审给出答案
论文阅读·人工智能·软件工程
字节数据平台6 分钟前
一客一策:Data Agent 如何重构大模型时代的智能营销
大数据·人工智能·重构
机器之心24 分钟前
字节Seedream 4.0将全量开放!抢先评测来了,我们摸索出AI生图一大波「邪修」玩法
人工智能·openai
空白到白27 分钟前
机器学习-集成学习
人工智能·机器学习·集成学习
水凌风里28 分钟前
4.4 机器学习 - 集成学习
人工智能·机器学习·集成学习
雲_kumo29 分钟前
集成学习:从理论到实践的全面解析
人工智能·机器学习·集成学习
用户51914958484538 分钟前
30条顶级APT与蓝队攻防单行命令:网络战场终极对决
人工智能·aigc
双向3338 分钟前
AI 辅助文档生成:从接口注释到自动化 API 文档上线
人工智能
算法打盹中1 小时前
基于树莓派与Jetson Nano集群的实验边缘设备上视觉语言模型(VLMs)的性能评估与实践探索
人工智能·计算机视觉·语言模型·自然语言处理·树莓派·多模态·jetson nano