[caffe(二)]Python加载训练caffe模型并进行测试2
16lz
2021-01-22
#coding=utf-8 import h5py import numpy as np import caffe #1.导入数据 filename = 'testdata.h5' f = h5py.File(filename, 'r') n1 = f.get('data') n1 = np.array(n1) print n1[0] n2=f.get( 'label_1d') n2 = np.array(n2) f.close() #2.导入模型与网络 deploy='gesture_deploy.prototxt' #deploy文件 caffe_model= 'iter_iter_1000.caffemodel' #训练好的 caffemodel net = caffe.Net(deploy,caffe_model,caffe.TEST) count=0 #统计预测值和标签相等的数量 t=1000 #t:样本的数量 for i in range(t): #数据处理 tempdata=n1[i,0:63] tempdata = np.reshape([[tempdata]], (1,1,63)) tempdata= tempdata.astype(np.float32) net.blobs['data'].data[0] = tempdata #预测 out = net.forward() output = out['outputs'] result= np.where(output==np.max(output)) predi=result[1][0] #判断predi与label是否相等,并统计 label = n2[i, 0] if predi==(label): count=count+1 kk=[predi,label] print kk print count
更多相关文章
- Bash脚本删除目录中多个文件名末尾的'x'字符数量?
- 配置 limits.conf 限制 Linux 用户登录数量
- JOIN vs. WHERE:为什么获得相同结果的两个查询显示3-4个数量级的
- 从分组中获取最大数量
- 记录不一致的数量-只改变是排序。
- 求sql存储过程!--创建存储过程统计各仓库各种物料的现存数量;
- Android实现桌面图标通知数量角标
- Java8 新特性之流式数据处理
- 转发可变数量的参数以通过椭圆运算符接受可变数量的参数