深度学习100问-13：深度学习如何制作个人数据集？

深度学习100问

Author：louwill

Machine Learning Lab

上一节我们谈到深度学习，尤其是计算机视觉方向的一些经典数据集。这些数据除了供我们学习和实验之外，在实际的生产环境下很难起到帮助作用。面对专业领域和定制化场景，我们往往需要自己动手采集和制作数据集。毕竟，有多少人工就有多少智能。数据量越大、数据标注质量越高，训练出来的AI模型才能够越强大。

本节笔者就介绍如何使用一些开源的标注工具来制作自己的深度学习训练数据集。数据标注是一项耗时和费力的工作，在一些专业领域，比如说医学图像的标注，可能标注起来难度更大、专业性更强。

目前开源的深度学习标注工具有很多。像Labelme、LabelImg、Vatic、Sloth、ImageJ、CVAT、Yolo_mark、RectLabel和Labelbox等等。本节就介绍如何使用开源的图像标注工具Labelme来制作个人的训练数据集。至于其它标注工具，大家可自行去搜索尝试。

CVAT语义分割标注

Labelme是MIT的计算机科学和人工智能实验室基于Python和Qt编写的一款开源图像标注工具。我们可以使用Labelme轻松实现分类、检测和分割等视觉任务的图像标注工作。

Labelme的安装非常简单，直接用Python的包安装工具pip即可安装：

pip install labelme

安装完成后直接在命令行输入labelme即可调出gui界面：

调出gui后，可以尝试打开某一张图片或者某一文件夹加载所有图片。打开任意一张图片，选择edit菜单栏下的create polygons进行手动标注，勾勒出每个动物的轮廓，最后点击保存为.json标签文件。

重新打开.json标注文件：

得到.json文件还不够，我们还需要将json转化为mask标签图像，直接在命令行调用labelme提供好的json_to_dataset函数将json进行标签转换：

labelme_json_to_dataset dogcat.json

转换完成后会生成一个标签文件夹，包括原始图像img.png、标签图像label.png、标签可视化图像label_viz.png、标签名称的.txt文件和.yaml格式标签名称文件。

查看label的像素分类值情况：

import numpy as np
from PIL import Image
np.unique(label)

array([0, 1, 2, 3], dtype=uint8)

labelme作为一个开源工具，我们也可以尝试对其源码进行修改或者二次开发来实现定制化的标注功能。labelme代码目录结构：

labelme在PASCAL VOC数据集上的实例分割标注示例：

参考资料：

https://github.com/wkentaro/labelme

往期精彩：

一个算法工程师的成长之路

深度学习100问-13：深度学习如何制作个人数据集？相关推荐

深度学习100问之深度学习的本质
今天在查找网上的BP博文([深度学习]backpropagation(反向传播))进行总结和撰写时发现了一篇博客和一个帖子引发了我的深刻思考,以下是分别的链接: Yes you should unde ...
深度学习100问-1：深度学习环境配置有哪些坑？
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达从今天起,开辟一个新的专栏,名字就叫深度学习100问.选取深度学习 ...
深度学习100问：图像语义分割有哪些经典的上采样方法？
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达大家都知道目前主流的语义分割模型都是编码-解码框架的.通过编码器不 ...
深度学习100问之深入理解Regularization（正则化）
声明 1)该文章整理自网上的大牛和机器学习专家无私奉献的资料,具体引用的资料请看参考文献. 2)本文仅供学术交流,非商用.所以每一部分具体的参考资料并没有详细对应.如果某部分不小心侵犯了大家的利益,还 ...
Java基础学习 100 问
Java基础学习 100 问来源: 赢在面试( [Java知音](javascript:void(0)
深度学习100问之深入理解Vanishing/Exploding Gradient（梯度消失/爆炸）
这几天正在看梯度消失/爆炸,在深度学习的理论中梯度消失/爆炸也是极其重要的,所以就抽出一段时间认真地研究了一下梯度消失/爆炸的原理,以下为参考网上的几篇文章总结得出的. 本文分为四个部分:第一部分主要 ...
深度学习100问：什么是深监督(Deep Supervision)？
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达所谓深监督(Deep Supervision),就是在深度神经网络 ...
pli测试50题题库_AI面试必备/深度学习100问1-50题答案解析
1.梯度下降算法的正确步骤,(正确步骤dcaeb)(梯度下降法其实是根据函数的梯度来确定函数的极小值),这里的问题与其说是梯度下降算法的步骤不如说类似图图像分类训练的整个流程:网络初始化-输入to输出 ...
深度学习100问之提高深度学习模型训练效果（调参经验）
声明 1)本文仅供学术交流,非商用.所以每一部分具体的参考资料并没有详细对应.如果某部分不小心侵犯了大家的利益,还望海涵,并联系博主删除. 2)博主才疏学浅,文中如有不当之处,请各位指出,共同进步,谢 ...

深度学习100问-13：深度学习如何制作个人数据集？

深度学习100问-13：深度学习如何制作个人数据集？相关推荐

最新文章

热门文章