手把手教你入门使用tf-slim库 | 回顾

tf-slim是基于tensorflow的高层封装库,包含了目前最新的reset-net,Google-Inception等网络的实现及图像处理算法,支持多GPU并行。使用tf-slim库,可以帮助你快速搭建图像处理(包括分类,分割)的生产环境,快速开发初代产品。本文内容为来自北京信息科技大学计算机应用技术专业研二在读生吴正一讲解的怎么快速入门tf-slim。

在雷锋网AI研习社公开课上,来自北京拓尔思研发部三部图像处理实习生吴正一为大家详细介绍了tf-slim库的入门使用知识、注意事项、以及显卡选购和机器配置。

吴正一,北京信息科技大学计算机应用技术专业研二在读。目前在北京拓尔思信息技术股份有限公司,担任研发三部图像处理实习生,负责专利图像检索系统开发;主要研究方向为计算机视觉与深度学习。

以下是他在AI研习社直播的分享内容整理:

大家好,我是来自北京信息科技大学计算机应用技术专业研二在读生吴正一,目前是拓尔思图像组实习。今天的分享主题是tf-slim怎样快速上手,主要讲基于TensorFlow的高层封装库tf-slim入门知识。

分享提纲

  • 装tensorflow-gpu,配置tf-slim环境
  • 使用tf-slim训练自己的数据,以及将原始图像转换成 .tfrecord 数据文件
  • 使用tf-slim加载tfrecord训练图像分类模型
  • 使用训练好的模型对单张图像分类
  • GPU的选购和机器配置等

tf-slim适用于快速处理工作上大型图像数据。之所以它能处理大型图像数据,是因为他有一个很好的数据处理过程,它能把原始的图像转换成 .tfrecord 队列格式,读取特别快,也更加稳定。

tf-slim 优点

  • 有最新的的网络实现和预训练模型,能够快速产出。谷歌把一些最新的,最重要的大型网络已经实现好了,封装在tf-slim里面,可直接调用。
  • 标准化的数据处理模板,适用于大型数据集。这个是比其他要好,假设处理一些数据,还要处理各种图像,这个提供了一套标准化数据处理,
  • 多GPU并行训练,快速产出
  • 基于TensorFlow,可实现精细化操控

原生TensorFlow框架可比作为汽车配件、钢板原材料,Keras 可比作发动机、车轮、车模型,Keras改源代码比较麻烦。tf-slim可比作一辆整车,但是又可以拆卸,因为它把最新的网络实现和预训练模型已经写好了,既封装了 一些网络实现,又能实现精细化的操作(可改源代码)。

tf-slim 配置

1、配置GPU版本的TensorFlow

2、加载tf-slim库

使用tf-slim训练自己的模型

以上具体参数可在文末视频中观看。实际上用的时候,不可能和网上一样,不可能一个文件跑到底。从训练最后一层开始,以及验证,然后再训练所有,再验证,分四个文件来跑。

视频中看scripts,可看到数据流动的过程。

将数据转换成 .tfrecord文件

训练模型

从第一个文件开始训练,第二个验证,第三个慢慢调参,第四个再验证。需要注意的是checkpoint文件的用法,它可以支持断点,如果训练停了之后,下一次它可以从最新的模型接着训练。如果不需要接着上一次继续训练,要把整个checkpoint文件删掉。

使用model分类测试图像

这里有一个坑需要注意一下,tf-slim 未提供单张图像的预测脚本,自己用的时候需要写一个单张图片加载模型,以及单张图片的分类。

选购GPU 的考虑及性价比对比,横坐标是性能,纵坐标为价格。

TensorFlow模型训练效率对比结果

待验证:1080 GPU在主机上的训练速度,双1080/1080Ti GPU的训练速度提升问题(修改TensorFlow网络架构)。

购买显卡,主板等硬件注意事项:

  • 主板需要支持SLI(专为Nividia显卡)双路连接技术。
  • 双显卡之间需要使用桥接器连接。如图一所示,分硬联和软联,硬联不可弯折。
  • 主板支持PCI-E*16插槽,装两张显卡需要至少两个PCI-E*16插槽(考虑到显卡占用较大物理空间,PCI-E*16插槽最好是和PCIE*1插槽交叉排列。)
  • 显卡电源线接在电源上,GTX1080等高端消费级显卡需要4*2pin的电源接口。

以上就是吴正一分享的主要内容。完整视频请点击阅读原文观看。

原文发布于微信公众号 - AI研习社(okweiwu)

原文发表时间:2017-12-09

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏AI研习社

手把手教你入门使用 tf-slim 库 | 回顾

tf-slim 是基于 tensorflow 的高层封装库,包含了目前最新的 reset-net,Google-Inception 等网络的实现及图像处理算法,...

4284
来自专栏深度学习与数据挖掘实战

【AI头条&优质资源】时间序列预测模型:使用深度神经网络RNN+Attention机制

放arxiv那天看了一下,整篇paper思路读下来还是非常清晰的,实验效果也很不错。

4522
来自专栏图形学与OpenGL

实验3 地理空间数据可视化

实验结果:(1)自己家乡及其位置信息;(2)图8-10,可以手绘,可以文字描述,也可以将图缩小打印贴上;

2682
来自专栏机器之心

业界 | 谷歌展示全新医疗诊断范式:深度学习+AR显微镜=实时检测癌症

选自Google Research Blog 作者:Martin Stumpe、Craig Mermel 机器之心编译 参与:路、刘晓坤 今天,谷歌在美国癌症研...

3406
来自专栏AI研习社

TensorFlow 全网最全学习资料汇总之TensorFlow的技术应用【4】

【AI研习社】关注AI前沿、开发技巧及技术教程等方面的内容。欢迎技术开发类文章、视频教程等内容投稿,邮件发送至:zhangxian@leiphone.com 随...

3307
来自专栏新智元

第一个手机端分布式深度学习系统,设计自动化顶会 DATE 最佳论文

【新智元导读】欧洲最大的设计自动化会议 DATE 昨天在瑞士洛桑召开。杜克大学陈怡然教授组关于移动平台深度学习计算的研究荣获嵌入式软件类最佳论文。陈怡然等人提出...

4868
来自专栏新智元

【AI防熊孩子抢手机】浙大开发算法软件,轻松识别儿童玩手机

新智元编译 来源:futurism、acm 编译:克雷格 【新智元导读】过年你的手机有没有被熊孩子抢走玩游戏?这个问题将来可能被算法攻破。由南卡罗来纳大...

36716
来自专栏大数据文摘

手把手|教你打造一个曲风分类机器人(附视频教程)

2784
来自专栏磐创AI技术团队的专栏

公交车总迟到?你大概掉进了“等待时间悖论"

你到了车站,准备搭乘声称每10分钟一班的公交车。你盯着你的手表留意着时间,结果公交车终于在11分钟后到来。

501
来自专栏AI研习社

如何利用机器学习预测房价?

对于我在梅蒂斯的最后一个项目,我希望能包含过去三个月里所学到的东西,而预测波特兰房价这个题目正符合我的要求,因为我能够将网络爬取技术、文本自然语言处理,图像上的...

39210

扫码关注云+社区

领取腾讯云代金券