【模型转换】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号线13 分钟前
机器学习笔试选择题:题组2
人工智能·算法·机器学习
算家云36 分钟前
OpenAI秘密测试ChatGPT安全路由,情感对话触发GPT-5严格审查
人工智能·chatgpt·算家云·openai秘密测试安全路由·算家计算·租算力,就到算家云
新加坡内哥谈技术36 分钟前
OpenAI近日推出了一项名为 ChatGPT Pulse 的全新功能
人工智能·chatgpt
hunteritself38 分钟前
DeepSeek V3.1-Terminus、阿里 Qwen3-Max、ChatGPT Pulse 同周登场!| AI Weekly 9.22-9.28
人工智能·ios·chatgpt·语音识别·iphone
ai_xiaogui41 分钟前
ChatGPT开源模型发布!部署20B/120B AI模型实战指南
人工智能·chatgpt·20b 120b模型部署教程·本地部署chatgpt模型实战·高性能加速开源ai模型开发·开源模型windows配置方法
春末的南方城市41 分钟前
阿里开源视频修复方法Vivid-VR:以独特策略与架构革新,引领生成视频修复高质量可控新时代。
人工智能·深度学习·机器学习·计算机视觉·aigc
源码部署21 小时前
Pytorch零基础入门到项目实战
pytorch
Juchecar1 小时前
人工智能重塑人类生活范式
人工智能
FIT2CLOUD飞致云1 小时前
飞致云开源社区月度动态报告(2025年9月)
人工智能·开源
小林z1 小时前
Pytorch零基础入门到项目实战
pytorch