博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
caffe——网络参数转化
阅读量:4322 次
发布时间:2019-06-06

本文共 1041 字,大约阅读时间需要 3 分钟。

在训练网络时可以利用别人的pre-train model来初始化的网络,caffe可以实现两个网络参数的转化,前提条件是转化的层的参数设计是一致的,以下程序是转化了三个卷积层和三个全连接层的参数,python的代码如下:

import caffecaffe.set_mode_gpu()train_net = caffe.Net('/home/python_code/caffe/trainmodel.prototxt',                      '/home/python_code/caffe/gendernet_50000.caffemodel', caffe.TEST)test_net = caffe.Net('/home/python_code/caffe/deploy.prototxt', caffe.TEST)test_net.save('/home/python_code/caffe/gendernet.caffemodel')params = ['conv1', 'conv2', 'conv3', 'fc6', 'fc7', 'fc8']params_trans = ['conv1', 'conv2', 'conv3', 'fc6', 'fc7', 'fc8']train_params = {pr: (train_net.params[pr][0].data, train_net.params[pr][1].data) for pr in params}test_params = {pr: (test_net.params[pr][0].data, test_net.params[pr][1].data) for pr in params_trans}for pr_train, pr_test in zip(params, params_trans):    test_params[pr_test][0].flat = train_params[pr_train][0].flat    test_params[pr_test][1][...] = train_params[pr_train][1]test_net.save('/home/python_code/caffe/gendernet.caffemodel')

 

转载于:https://www.cnblogs.com/leo-sun/p/6159769.html

你可能感兴趣的文章
IE浏览器清浮动
查看>>
Ubuntu docker 使用命令 系列二
查看>>
在SharePoint 2010中使用托管元数据导航和关键词筛选器查找列表或库的内容
查看>>
学习MongoDB(二) Replica Set集群配置
查看>>
在只有一个网线的前提下,实现两个电脑之间的局域网通信(伽卡他卡电子教室通信)...
查看>>
maximun-depth-of-binary-tree
查看>>
jump game
查看>>
pose项目里我遇到的问题
查看>>
JNDI 详解
查看>>
html5 本地存储
查看>>
C#中的static、readonly与const的比较
查看>>
理解linux 块, i节点
查看>>
元素垂直或居中定位
查看>>
Selenium自动化-调用Mysql数据库
查看>>
项目一
查看>>
Framework/base 下添加自定义模块的步骤
查看>>
[转载]AAF灵便应用框架简介系列(6):休息一下,泛谈面向对象 Why OO+多层结构?...
查看>>
android EditView ime
查看>>
关于OpenXml SpreadSheet列宽根据内容的Auto-suitability
查看>>
javascript 学习随笔7
查看>>