摘要: import numpy as np import caffe import os f = open('/home/python_code/result.txt', 'wr') caffe.set_mode_gpu() net = caffe.Net('/home/python_code/caffe_predict/deploy.prototxt', '/ho... 阅读全文
posted @ 2016-12-11 16:08 拖拉机的音响 阅读(2503) 评论(1) 推荐(0)
摘要: 在训练网络时可以利用别人的pre-train model来初始化的网络,caffe可以实现两个网络参数的转化,前提条件是转化的层的参数设计是一致的,以下程序是转化了三个卷积层和三个全连接层的参数,python的代码如下: 阅读全文
posted @ 2016-12-11 15:58 拖拉机的音响 阅读(950) 评论(0) 推荐(0)