caffe python接口 测试分类:
来源:互联网 发布:apt-get insall yum 编辑:程序博客网 时间:2024/05/24 04:33
原文链接
#coding=utf-8#加载必要的库import numpy as npimport sys,os#设置当前目录caffe_root = '/home/xxx/caffe/' sys.path.insert(0, caffe_root + 'python')import caffeos.chdir(caffe_root)net_file=caffe_root + 'models/bvlc_reference_caffenet/deploy.prototxt'caffe_model=caffe_root + 'models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel'mean_file=caffe_root + 'python/caffe/imagenet/ilsvrc_2012_mean.npy'net = caffe.Net(net_file,caffe_model,caffe.TEST)transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})transformer.set_transpose('data', (2,0,1))transformer.set_mean('data', np.load(mean_file).mean(1).mean(1))transformer.set_raw_scale('data', 255) transformer.set_channel_swap('data', (2,1,0))im=caffe.io.load_image(caffe_root+'examples/images/cat.jpg')net.blobs['data'].data[...] = transformer.preprocess('data',im)out = net.forward()imagenet_labels_filename = caffe_root + 'data/ilsvrc12/synset_words.txt'labels = np.loadtxt(imagenet_labels_filename, str, delimiter='\t')top_k = net.blobs['prob'].data[0].flatten().argsort()[-1:-6:-1]for i in np.arange(top_k.size): print top_k[i], labels[top_k[i]]
阅读全文
0 0
- caffe python接口 测试分类:
- python接口测试caffe模型
- caffe-python接口图片分类demo
- python接口调用已训练好的caffe模型测试分类
- caffe python接口:用训练好的模型来分类
- ubuntu下caffe手写数字识别问题、python接口测试
- Caffe+Python接口测试MNIST--下载和准备数据
- 使用caffe的Python接口进行训练测试
- 【Caffe的C++接口使用说明六)】caffe中分类接口C++源代码的再次解读以及测试文件的完善
- Caffe windows下入门级别的从编译到训练然后到分类(用python接口)测试自己的图片数据(四)
- Caffe的Python接口
- caffe python 接口使用
- Caffe的Python接口
- caffe---python接口配置
- Caffe Python接口编译
- caffe python接口编译
- caffe python接口:mnist
- caffe配置python接口
- (四) java并发编程--线程安全和资源共享
- 欢迎使用CSDN-markdown编辑器
- OpenStack虚拟机创建流程
- springMVC解决跨域jsonp乱码问题的三种方法
- jvm内存模型
- caffe python接口 测试分类:
- 用一个例子看const与指针、引用、多级指针结合的区别
- 机器学习笔记(三) 线性代数复习
- git版本冲突及解决方案
- MMC 卡驱动分析
- Java基础知识梳理--泛型
- Python10--面向对象
- 数组
- POJ