博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
8-3 下载Google图像识别网络inception-v3并查看结构
阅读量:5957 次
发布时间:2019-06-19

本文共 1866 字,大约阅读时间需要 6 分钟。

import tensorflow as tfimport osimport tarfileimport requests# inception-v3 是googlenet的第三个版本#inception模型下载地址#inception_pretrain_model_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz' #这里采用手动下载后直接放入下述模型存放地址中。 #模型存放地址,#inception_pretrain_model_dir = "inception_model"   #此文件夹如果不存在会自动创建if not os.path.exists(inception_pretrain_model_dir):    os.makedirs(inception_pretrain_model_dir)    #获取文件名,以及文件路径filename = inception_pretrain_model_url.split('/')[-1]filepath = os.path.join(inception_pretrain_model_dir, filename)#下载模型if not os.path.exists(filepath):    print("download: ", filename)    r = requests.get(inception_pretrain_model_url, stream=True)    with open(filepath, 'wb') as f:        for chunk in r.iter_content(chunk_size=1024):            if chunk:                f.write(chunk)print("finish: ", filename)#解压文件tarfile.open(filepath, 'r:gz').extractall(inception_pretrain_model_dir) #模型结构存放文件log_dir = 'inception_log'if not os.path.exists(log_dir):    os.makedirs(log_dir)#classify_image_graph_def.pb为google训练好的模型inception_graph_def_file = os.path.join(inception_pretrain_model_dir, 'classify_image_graph_def.pb')#'classify_image_graph_def.pb'为inception-v3中训练好的一个模型with tf.Session() as sess:    #创建一个图来存放google训练好的模型    with tf.gfile.FastGFile(inception_graph_def_file, 'rb') as f:        graph_def = tf.GraphDef()        graph_def.ParseFromString(f.read())        tf.import_graph_def(graph_def, name='')    #保存图的结构    writer = tf.summary.FileWriter(log_dir, sess.graph)    writer.close()

运行结果输出:

finish:  inception-2015-12-05.tgz
并在inception_model文件夹中产生了如下文件:

 

在inception_log文件夹中生成如下文件:

 在cmd中打开tensorboard:

在chrome中打开localhost:6006,得到GRAPHS:

inception-v3中最具特色的时mixed层:

 

 mixed层中有并排四个通道:一个卷积层,两个卷积层,三个卷积层,一个池化层加一个卷积层。

四个并排增加模型的宽度,好几个层串联叠加增加了模型的深度。

也把mixed的结构称为inception结构

 

 

 

转载于:https://www.cnblogs.com/Josie-chen/p/9065880.html

你可能感兴趣的文章
中国人工智能学会通讯——自然语言处理中的技术评测
查看>>
开启openssl
查看>>
你必须关注超融合基础设施的理由
查看>>
善用佳软站长:畅谈大数据时代的知识管理
查看>>
AT&T:BYOD在2015年达“拐点”
查看>>
一款高端精密的DDoS定制工具包
查看>>
甲骨文5000万美元收购以色列大数据公司Crosswise
查看>>
小程序技术方案探讨
查看>>
用友优普发布企业空间2.0 助推企业互联网+
查看>>
文思海辉:大数据发展—源于创新、服务于创新
查看>>
中产委廖明:给中国制造转型的三点建议
查看>>
记一场为未来就绪的企业客户峰会
查看>>
5G来了 中国移动将孵化各种垂直应用 实现万物互联
查看>>
Fortinet为企业提出应对APT攻击的一些建议
查看>>
使用Disk2VHD进行P2V转换需要知道的一些事
查看>>
PHP图片处理库Grafika详细教程(2):图像特效处理模块
查看>>
LXD 2.0 系列(八):LXD中的LXD
查看>>
安装WMware 在Windows平台下学习Linux
查看>>
NodeJS对于Java开发者而言是什么?
查看>>
2016 软件开发的七大趋势:容器技术将统治世界
查看>>