caffe 红绿灯识别
来源:互联网 发布:石油经济评价软件 编辑:程序博客网 时间:2024/04/29 11:51
#coding=utf-8 #加载必要的库 import numpy as np import sys,os #设置当前目录 caffe_root = '/home/ubuntu/caffe/' sys.path.insert(0, caffe_root + 'python') import caffe os.chdir(caffe_root) net_file='/home/ubuntu/Downloads/deep-learning-traffic-lights-master/model/deploy.prototxt' caffe_model='/home/ubuntu/Downloads/deep-learning-traffic-lights-master/model/train_squeezenet_scratch_trainval_manual_p2__iter_8000.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('/home/ubuntu/Downloads/deep-learning-traffic-lights-master/4.jpg') net.blobs['data'].data[...] = transformer.preprocess('data',im) out = net.forward() imagenet_labels_filename = '/home/ubuntu/Downloads/deep-learning-traffic-lights-master/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]]
synset_words.txt
yelloredgreen
阅读全文
0 0
- caffe 红绿灯识别
- caffe 红绿灯识别
- Arduino寻迹与识别红绿灯
- 红绿灯
- 红绿灯
- Caffe-图像识别训练
- caffe 人脸识别
- caffe识别手写数字
- caffe-车尾识别
- CAFFE 人脸表情识别
- Caffe学习-手写数字识别
- Caffe学习-手写数字识别
- Caffe学习-手写数字识别
- Caffe学习-手写数字识别
- 关于红绿灯
- 红绿灯汇编程序
- ABAP红绿灯
- 红绿灯-DEMO_DYNPRO_STATUS_ICONS
- CmakeList 组织编译项目的基本用法
- CodeVS2102 石子归并 2 解题报告【区间DP】
- HDU 2828 Lamp DLX重复覆盖
- 2017ccps网络赛 1003 Friend-Graph(暴力)HDU 6152
- 替换对话框
- caffe 红绿灯识别
- 简单操作 Cookie 注册登录
- IO流
- 莫比乌斯反演-HDU1695
- JAVA操作selenium的简单示例
- poj Friendship 最小割+拆点
- Shell好的学习资料
- Sort List
- caffe源码解析—caffe layer的工作原理理解