#?-*-?coding:?utf-8?-*-
#?@Time????:?8/1/18?10:39?AM
#?@Author??:?‘xxguo‘
#?@File????:?test_classify.py
import?numpy?as?np
import?sys
import?requests
from?kaptcha?import?_chr_code_ascll

#ccaffe路徑文件
caffe_root?=?‘/home/xxguo/Project/caffe‘

sys.path.insert（0?caffe_root?+?‘/python‘）
import?os
os.environ[‘GLOG_minloglevel‘]?=?‘1‘
import?caffe
size?=?256
#?項目文件目錄
project_root?=?‘/xx/‘

model_def?=?project_root?+?‘models/deploy.prototxt‘
model_weights?=?project_root?+?‘media/test/solver_iter_1000.caffemodel‘

net?=?caffe.Net（model_def?model_weights?caffe.TEST）
mu?=?np.load（caffe_root?+?‘/python/caffe/imagenet/ilsvrc_2012_mean.npy‘）
mu?=?mu.mean（1）.mean（1）
transformer?=?caffe.io.Transformer（{‘data‘:?net.blobs[‘data‘].data.shape}）
transformer.set_transpose（‘data‘?（201））
transformer.set_mean（‘data‘?mu）
transformer.set_raw_scale（‘data‘?255）
transformer.set_channel_swap（‘data‘?（210））
net.blobs[‘data‘].reshape（13size?size）


def?code_verification（image_file_path?code=None）:
????image?=?caffe.io.load_image（image_file_path）
????net.blobs[‘data‘].data[...]?=?transformer.preprocess（‘data‘?image）
????caffe.set_mode_cpu（）
????output?=?net.forward（）
????output_prob_1?=?output[‘prob_1‘][0].argmax（）??#?給出概率最高的是第幾類，需要自己對應到我們約定的類別去
????output_prob_2?=?output[‘prob_2‘][0].argmax（）
????output_prob_3?=?output[‘prob_3‘][0].argmax（）
????output_prob_4?=?output[‘prob_4‘][0].argmax（）
????print?output_prob_1?output_prob_2?output_prob_3?output_prob_4
????prob1?=?output[‘prob_1‘][0].flatten（）
????order1=prob1.argsort（）[::-1][:5]
????print?zip（prob1[order1]?map（_chr_code_ascll?order1））
????prob2?=?output[‘prob_2‘][0].flatten（）
????order2=prob2.argsort（）[::-1][:5]
????print?zip（prob2[order2]?map（_chr_code_ascll?order2））
????prob3?=?output[‘prob_3‘][0].flatten（）
????order3=prob3.argsort（）[::-1][:5]
????print?zip（prob3[order3]?map（_chr_code_ascll?order3））
????prob4?=?output[‘prob_4‘][0].flatten（）
????order4=prob4.argsort（）[::-1][:5]
????print?zip（prob4[order4]?map（_chr_code_ascll?order4））
????pro_code?=?‘‘.join（map（_chr_code_ascll?[output_prob_1?output_prob_2?output_prob_3?output_prob_4]））
????if?code:
????????if?code?==?pro_code:
????????????print?‘預測成功:{0}‘.format（pro_code）
????????else:
????????????print?‘預測失敗code:{0}?pro_code:{1}‘.format（code?pro_code）
????else:
????????print?‘本次預測值為：{0}‘.format（pro_code）

def?get_image_path（image_pat?code）:
	“““
	測試圖片路徑
	“““
	image_pat?=?image_pat
	code?=?code
	return?image_pat?code

for?i?in?range（1）:
????print?‘--------------start----------------start‘
????image_path?code?=?get_image_path（‘./test/pro_64db.jpg‘?‘64db‘）
????code_verification（image_path?code）
????print?‘===========end====================end‘

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-08-01?18:03??通用圖片驗證碼識別（caffe使用案例教程）\
?????文件??????523554??2018-07-06?10:57??通用圖片驗證碼識別（caffe使用案例教程）\kaptcha.war
?????目錄???????????0??2018-08-01?17:46??通用圖片驗證碼識別（caffe使用案例教程）\models\
?????文件????????7956??2018-08-01?17:46??通用圖片驗證碼識別（caffe使用案例教程）\models\deploy.prototxt
?????文件?????????338??2018-08-01?17:46??通用圖片驗證碼識別（caffe使用案例教程）\models\solver.prototxt
?????文件????????9640??2018-08-01?17:46??通用圖片驗證碼識別（caffe使用案例教程）\models\train_val.prototxt
?????目錄???????????0??2018-08-01?17:45??通用圖片驗證碼識別（caffe使用案例教程）\python\
?????目錄???????????0??2018-08-01?17:45??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\
?????文件????????3546??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\classifier.py
?????文件????????3366??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\classifier.pyc
?????文件????????6721??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\coord_map.py
?????文件????????8549??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\detector.py
?????文件????????7489??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\detector.pyc
?????文件???????11174??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\draw.py
?????目錄???????????0??2018-08-01?17:45??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\imagenet\
?????文件?????1572944??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\imagenet\ilsvrc_2012_mean.npy
?????文件???????13079??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\io.py
?????文件???????13716??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\io.pyc
?????文件????????8277??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\net_spec.py
?????文件???????10646??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\net_spec.pyc
?????目錄???????????0??2018-08-01?17:45??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\proto\
?????文件??????245741??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\proto\caffe_pb2.py
?????文件??????124401??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\proto\caffe_pb2.pyc
?????文件???????????0??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\proto\__init__.py
?????文件?????????142??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\proto\__init__.pyc
?????文件???????11615??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\pycaffe.py
?????文件???????12685??2018-07-24?16:51??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\pycaffe.pyc
?????目錄???????????0??2018-08-01?17:45??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\test\
?????文件????????6894??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\test\test_coord_map.py
?????文件????????1114??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\test\test_draw.py
?????文件????????1694??2018-07-05?17:26??通用圖片驗證碼識別（caffe使用案例教程）\python\caffe\test\test_io.py
............此處省略22個文件信息