caffe 做测试的方法

最新推荐文章于 2021-10-22 16:12:00 发布

scarlette47

最新推荐文章于 2021-10-22 16:12:00 发布

阅读量375

点赞数

分类专栏：机器学习 caffe

本文链接：https://blog.csdn.net/u011037837/article/details/103249867

版权

本文介绍了如何使用Caffe进行图像分类的测试，包括通用方法、针对单张图片和多张图片的分类测试。通用方法涉及调用Python接口和命令行形式；分类问题测试中，展示了使用Python的caffe.Classifier类以及通过.sh文件或命令行运行classification.bin。对于多张图片，建议修改classification.cpp以实现批量分类。

摘要由CSDN通过智能技术生成

（一）通用方法

1、调用python接口

net = caffe.Net(net_file,caffe_model,caffe.TEST)
net.blobs['data'].data[...] = transformer.preprocess('data', im)

net.forword

"""
caffe_test.py
http://www.lewisjin.coding.me
~~~~~~~~~~~~~~~
This script implement by Jin Fagang.
: copyright: (c) 2017 Didi-Chuxing.
: license: Apache2.0, see LICENSE for more details.
"""
import numpy as np
import sys
import os
import cv2

caffe_root = '/home/jfg/caffe/'
sys.path.insert(0, caffe_root + 'python')
import caffe


net_file = '/home/jfg/Documents/PythonSpace/caffe_tiny5/solver/caffenet_deploy.prototxt'
caffe_model = '/home/jfg/Documents/PythonSpace/caffe_tiny5/solver/model/caffenet_iter_4500.caffemodel'
mean_file = '/home/jfg/Documents/PythonSpace/caffe_tiny5/data/caffe_mean.bin