TensorFlow 之基于Inception V3的多标签分类 retrain
本文参考http://blog.csdn.net/Numeria/article/details/73604339
以及参考开源代码github链接: https://github.com/BartyzalRadek/Multi-label-Inception-net
一、准备训练数据
1.下载数据集
本文采用南京大学开源的数据集(点击下载:http://lamda.nju.edu.cn/files/miml-image-data.rar)
数据集中含有2000张图像,5个类,分别为 desert, mountains, sea, sunset , trees。
下载后包含两个压缩文件包:original.rar 以及 processed.rar
original.rar 中包含了2000张图像数据
示例如下:
processed.rar 中包含了图像对应的标签信息。
解压后为matlab文件格式。
本文需要的是标签矩阵 target.mat
示例如下:
2.转换标签文件
本文需将 target.mat 转换为txt格式的标签文件。
对于每一张图像 imagename.jpg, 对应生成一个 imagename.jpg.txt 的标签文件,文件中每一行为图像所含标签。
对于下图左边的图像,有标签 desert,mountains,则对应的标签文件内容如右边所示。
对于 target.mat 文件,转化的matlab脚本如下:
load('miml data.mat');
if ~exist('labeldir') mkdir labeldir;
end
LabelDir='labeldir/';
for i = 1:2000stri = num2str(i);label_file_name = [LabelDir stri '.jpg.txt'];fid = fopen(label_file_name,'w');for j = 1:5if targets(j,i)==1fprintf(fid,'%s\n',class_name{j});endendfclose(fid);
end
处理后的文件为:https://pan.baidu.com/s/1pLoeC6R
二、多标签分类
本文利用已训练好的权重,将最后一层替换为自定义的数据集的类别数,前面权值不变,只训练最后的一层。
1.将包含所有图像的文件夹放入 Multi-label-Inception-net/images 目录下
2.将labeldir目录拷贝到 Multi-label-Inception-net 目录下,改名为image_labels_dir(替换原同名目录)
3.运行 python 脚本
python retrain.py \
--bottleneck_dir=bottlenecks \
--how_many_training_steps 500 \
--model_dir=model_dir \
--output_graph=retrained_graph.pb \
--output_labels=retrained_labels.txt \
--summaries_dir=retrain_logs \
--image_dir=images
开始训练
训练完成
三、图像预测
利用脚本label_image.py 进行图像预测
调用如下:
python label_image.py images/image/4.jpg
得到输出结果:
TensorFlow 之基于Inception V3的多标签分类 retrain相关推荐
- 基于sigmoid的文本多标签分类模型代码实现
sigmoid一般是用来做二分类的,它是将一个标量的数字转换成[0,1]之间的一个概率值,如果概率值大于0.5, 则判定为是某个分类,否则则不是某个分类,公式如下: 本文基于sigmoid做了一个将一 ...
- 条件随机场python实现_基于条件随机场的多标签分类
基本的CRF不支持多标签分类.然而,一些扩展已经被探索,例如集合多标签(CML)和 多标签(CMLF).从(1):A conditional random field (CRF) based mode ...
- 多标签分类方法总结——实现方法、评价指标、损失函数
目录 一.两种思想总结 1.问题转换 2.算法改编 二.深度网络多标签分类 三.多标签分类评价指标 四.多标签分类的损失函数 1.二分类和多分类 2.多标签分类 五.参考文章 注:本文为总结性文章,应 ...
- win10使用Inception v3进行图像分类TensorFlow学习记录
win10中构建TensorFlow环境: 下载并安装anaconda环境,具体步骤这里有:spark2.3在Windows10当中来搭建python3的使用环境pyspark,只需要看anacond ...
- Tensorflow(七)Retrain Google Inception V3
1.下载Inception V3模型 Download-Link 在tensorflow官网中可以直接下载,下载完压缩包以后解压,注意不要删除这个压缩包,后面可能会用到,然后在同目录下创建一个log文 ...
- 【TensorFlow系列】【五】利用inception v3 pb模型文件做预测
2019独角兽企业重金招聘Python工程师标准>>> 本文介绍如何利用imagenet比赛上训练好的inception v3冻结的pb模型进行inference. 1.下载ince ...
- 【Tensorflow】深度学习实战05——Tensorflow实现Inception V3
[fishing-pan:https://blog.csdn.net/u013921430转载请注明出处] 前言 前些日子在忙其他的事情,一直没有更新自己学习神经网络的博客,就在端午这天更吧!也祝大家 ...
- Inception V3 的 tensorflow 实现
tensorflow 官方给出的实现:models/inception_v3.py at master · tensorflow/models · GitHub 1. 模型结构 首先来看 Incept ...
- tensorflow实现inception V3
一.Inception V3结构介绍 1.Inception网络结构 在上图中下一层的输入是上一层的输出,在Inception V3中采用下面两种结构来减少图片的尺寸 在Inception V3中,卷 ...
最新文章
- 1476. Lunar Code
- [operator]ubuntu + git
- 用python tkinter显示Mandelbrot图
- 20个数据库设计的最佳实践
- pl/sql里的exists和in的差别
- leetcode :动态规划
- Go vs .NET Core 2.1
- jar注册成服务器,Spring Boot项目生成jar包,并在windows服务器中注册成服务,开机启动(示例代码)...
- 百试不爽的求爱技巧百试不爽的求爱技巧
- Jenkins构建jmeter进行接口测试的一些配置
- layout elements
- 受控文本生成任务简述
- 【Firewalld(Iptables)】
- java after 函数_函数周期表丨信息丨值丨ISONORAFTER
- this installer requires you to restart your system to finish installing Microsoft VC
- spring封装VerifyCodeUtil工具类,生成图片验证码
- CIC滤波器原理及设计实现
- Linux京东签到教程,京东POP店铺签到有礼操作指南
- 2022年合肥及合肥市各地稳岗补贴申请条件及补贴标准申请流程实施细则
- idea中按住ctrl+滚轮改变字体大小
热门文章
- python rgb转lab_RGB转LAB色彩空间
- 出现 I/Choreographer: Skipped 160 frames! The application may be doing too much work on its main thre
- 公司企业邮箱怎么注册?登录企业邮箱使用须知
- 无线射频专题《射频合规,2.4GHz WIFI信道分布与国家限制》
- 指标异动排查中,如何量化对大盘的贡献程度
- Spring Cloud Ribbon 负载均衡策略
- 【嵌入式知识12】使用PWM实现LED呼吸灯
- 读书感受 之 《一本书读懂金融常识》
- 使用UltraEdit比较文件的异同
- ATFX美指:小非农ADP数据来袭,美国是否出现经济衰退的关键指标