此篇文章主要是给大家介绍了python深度神经网络tensorflow练习好一点的实体模型开展图像分类实例详细说明,感兴趣的小伙伴可以参考借鉴一下,希望可以有一定的帮助,祝愿大家多多的发展,尽早涨薪。
文章正文
Google在各类图象数据库系统ImageNet上练习好啦1个Inception-v3实体模型,这一实体模型大家也可以用来进去图像分类。
下载地址:https://pan.baidu.com/s/1XGfwYer5pIEDkpM3nM6o2A
提取码:hu66
免费下载完缓解压力后,获得好多个文档:
在其中
classify_image_graph_def.pb文件便是练习好一点的Inception-v3实体模型。
imagenet_synset_to_human_label_map.txt是类型文档。
随机找一张图片
对这张图片进行识别,看它属于什么类?
代码如下:先创建一个类NodeLookup来将softmax概率值映射到标签上。
然后创建一个函数create_graph()来读取模型。
读取图片进行分类识别
</>复制代码
#-*-coding:utf-8-*-
import tensorflow as tf
import numpy as np
import re
import os
model_dir='D:/tf/model/'
image='d:/cat.jpg'
#将类别ID转换为人类易读的标签
class NodeLookup(object):
def __init__(self,
label_lookup_path=None,
uid_lookup_path=None):
if not label_lookup_path:
label_lookup_path=os.path.join(
model_dir,'imagenet_2012_challenge_label_map_proto.pbtxt')
if not uid_lookup_path:
uid_lookup_path=os.path.join(
model_dir,'imagenet_synset_to_human_label_map.txt')
self.node_lookup=self.load(label_lookup_path,uid_lookup_path)
def load(self,label_lookup_path,uid_lookup_path):
if not tf.gfile.Exists(uid_lookup_path):
tf.logging.fatal('File does not exist%s',uid_lookup_path)
if not tf.gfile.Exists(label_lookup_path):
tf.logging.fatal('File does not exist%s',label_lookup_path)
#Loads mapping from string UID to human-readable string
proto_as_ascii_lines=tf.gfile.GFile(uid_lookup_path).readlines()
uid_to_human={}
p=re.compile(r'[nd]*[S,]*')
for line in proto_as_ascii_lines:
parsed_items=p.findall(line)
uid=parsed_items[0]
human_string=parsed_items[2]
uid_to_human[uid]=human_string
#Loads mapping from string UID to integer node ID.
node_id_to_uid={}
proto_as_ascii=tf.gfile.GFile(label_lookup_path).readlines()
for line in proto_as_ascii:
if line.startswith('target_class:'):
target_class=int(line.split(':')[1])
if line.startswith('target_class_string:'):
target_class_string=line.split(':')[1]
node_id_to_uid[target_class]=target_class_string[1:-2]
#Loads the final mapping of integer node ID to human-readable string
node_id_to_name={}
for key,val in node_id_to_uid.items():
if val not in uid_to_human:
tf.logging.fatal('Failed to locate:%s',val)
name=uid_to_human[val]
node_id_to_name[key]=name
return node_id_to_name
def id_to_string(self,node_id):
if node_id not in self.node_lookup:
return''
return self.node_lookup[node_id]
#读取训练好的Inception-v3模型来创建graph
def create_graph():
with tf.gfile.FastGFile(os.path.join(
model_dir,'classify_image_graph_def.pb'),'rb')as f:
graph_def=tf.GraphDef()
graph_def.ParseFromString(f.read())
tf.import_graph_def(graph_def,name='')
#读取图片
image_data=tf.gfile.FastGFile(image,'rb').read()
#创建graph
create_graph()
sess=tf.Session()
#Inception-v3模型的最后一层softmax的输出
softmax_tensor=sess.graph.get_tensor_by_name('softmax:0')
#输入图像数据,得到softmax概率值(一个shape=(1,1008)的向量)
predictions=sess.run(softmax_tensor,{'DecodeJpeg/contents:0':image_data})
#(1,1008)->(1008,)
predictions=np.squeeze(predictions)
#ID-->English string label.
node_lookup=NodeLookup()
#取出前5个概率最大的值(top-5)
top_5=predictions.argsort()[-5:][::-1]
for node_id in top_5:
human_string=node_lookup.id_to_string(node_id)
score=predictions[node_id]
print('%s(score=%.5f)'%(human_string,score))
sess.close()
最后输出
tiger cat(score=0.40316)
Egyptian cat(score=0.21686)
tabby,tabby cat(score=0.21348)
lynx,catamount(score=0.01403)
Persian cat(score=0.00394)
综上所述,上述就给大家介绍完毕了,希望可以给大家带来帮助。
文章版权归作者所有,未经允许请勿转载,若此文章存在违规行为,您可以联系管理员删除。
转载请注明本文地址:https://www.ucloud.cn/yun/128739.html
本文主要是给大家介绍了caffe的python插口生成deploy文件学习培训及其用练习好一点的实体模型(caffemodel)来归类新的图片实例详细说明,感兴趣的小伙伴可以参考借鉴一下,希望可以有一定的帮助,祝愿大家多多的发展,尽早涨薪 caffe的python插口生成deploy文件 假如要将练习好一点的实体模型用于检测新的图片,那必然必须得一个deploy.prototxt文件,这一...
阅读 989·2023-01-14 11:38
阅读 987·2023-01-14 11:04
阅读 824·2023-01-14 10:48
阅读 2243·2023-01-14 10:34
阅读 1052·2023-01-14 10:24
阅读 956·2023-01-14 10:18
阅读 595·2023-01-14 10:09
阅读 648·2023-01-14 10:02