python接口调用已训练好的caffe模型测试分类方法

作者:Baterforyou 时间:2021-08-11 18:31:04 

训练好了model后,可以通过python调用caffe的模型,然后进行模型测试的输出。

本次测试主要依靠的模型是在caffe模型里面自带训练好的结构参数:~/caffe/models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel,以及结构参数

:~/caffe/models/bvlc_reference_caffenet/deploy.prototxt相结合,用python接口进行调用。

训练的源代码以及相应的注释如下所示:


# -*- coding: UTF-8 -*-
import os
import caffe
import numpy as np
root='/home/zf/caffe/'#指定根目录
deploy=root+'models/bvlc_reference_caffenet/deploy.prototxt'#结构文件
caffe_model=root + 'models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel'
#已经训练好的model

dir =root+'examples/images/'#保存测试图片的集合
filelist=[]
filenames=os.listdir(dir)
for fn in filenames:
 fullfilename = os.path.join(dir,fn)
 filelist.append(fullfilename)
#filelist.append(fn)
def Test(img):
#加载模型
 net = caffe.Net(deploy,caffe_model,caffe.TEST)

# 加载输入和配置预处理
 transformer = caffe.io.Transformer({'data':net.blobs['data'].data.shape})
 transformer.set_mean('data', np.load('/home/zf/caffe/python/caffe/imagenet/ilsvrc_2012_mean.npy').mean(1).mean(1))
 transformer.set_transpose('data', (2,0,1))
 transformer.set_channel_swap('data', (2,1,0))
 transformer.set_raw_scale('data', 255.0)

#注意可以调节预处理批次的大小
#由于是处理一张图片,所以把原来的10张的批次改为1
 net.blobs['data'].reshape(1,3,227,227)

#加载图片到数据层
 im = caffe.io.load_image(img)
 net.blobs['data'].data[...] = transformer.preprocess('data', im)

#前向计算
 out = net.forward()

# 其他可能的形式 : out = net.forward_all(data=np.asarray([transformer.preprocess('data', im)]))

#预测分类
 print out['prob'].argmax()

#打印预测标签
 labels = np.loadtxt("/home/zf/caffe/data/ilsvrc12/synset_words.txt", str, delimiter='\t')
 top_k = net.blobs['prob'].data[0].flatten().argsort()[-1]
 print 'the class is:',labels[top_k]
 f=file("/home/zhengfeng/caffe/examples/zf/label.txt","a")
 f.writelines(img+' '+labels[top_k]+'\n')
labels_filename=root +'data/ilsvrc12/synset_words.txt'
#循环遍历文件夹root+'examples/images/'下的所有图片
for i in range(0,len(filelist)):
 img=filelist[i]
 Test(img)

ps:主要有以下的文件需要说明

待测试的文件夹里面的图片数据为:

python接口调用已训练好的caffe模型测试分类方法

最后的输出结果如下:

以下是本人定义的label.txt文件写入的预测的数据:

python接口调用已训练好的caffe模型测试分类方法

如果在编译的时候出现import caffe error的话,说明没有导入caffe

Export PYTHONPATH=$PYTHONPATH:/home/zf/caffe/python,如果还是不行,可能是你的caffe的python接口未编译,cd /home/zf/caffe,然后执行make pycaffe,接着再测试。

来源:https://blog.csdn.net/baterforyou/article/details/71430284

标签:python,caffe,模型,分类
0
投稿

猜你喜欢

  • php短视频源码之瀑布流分割线左右间距问题及解决代码

    2023-06-12 00:27:07
  • 深入浅出SQL之左连接、右连接和全连接

    2009-08-30 15:14:00
  • css利用锚点控制翻页

    2008-07-17 13:12:00
  • 一段ASP的HTTP_REFERER判断代码

    2011-03-25 10:41:00
  • 使用Python从零开始撸一个区块链

    2021-06-02 07:28:22
  • AJAX初体验之上手篇

    2007-08-23 08:50:00
  • python通过加号运算符操作列表的方法

    2023-11-12 13:44:04
  • Javascript 每日测试 - 第五期 callee及function

    2008-07-10 13:22:00
  • Javascript DOM 编程实例讲解--仿LightBox效果提示框

    2008-05-01 13:25:00
  • 精美漂亮的php分页类代码

    2023-11-22 08:28:09
  • SQL Server中的执行引擎入门 图解

    2012-06-06 20:08:26
  • Rs.Open参数说明

    2008-05-12 22:43:00
  • php的优点总结 php有哪些优点

    2023-11-14 16:03:25
  • ORACLE8的分区管理

    2023-07-13 14:42:43
  • asp如何自动更新导航栏?

    2010-07-07 12:10:00
  • ASP中从数据库读取二进制文件数据代码

    2010-04-24 15:44:00
  • 《CSS禅意花园》学习笔记

    2008-10-20 12:43:00
  • 用python删除文件夹中的重复图片(图片去重)

    2021-07-23 22:42:36
  • Asp中如何设计跨越域的Cookie

    2008-10-24 09:46:00
  • 利用canvas实现的加载动画效果实例代码

    2023-06-29 22:49:51
  • asp之家 网络编程 m.aspxhome.com