要实现深度学习图像分割应用,首先要获取图像分割标注数据,比如PASCAL VOC、COCO、SBD等大型数据集,但这些数据集主要用于训练预训练模型和评价分割模型精度性能,针对实际应用还需要我们根据项目需求制作自己的训练数据集。这里介绍一种基于labelme图像标注工具的图像分割数据集制作流程。
一、安装图像标注工具labelme
pip install labelme==3.6
安装完成后在终端输入命令labelme,如果出现标注软件界面,则证明安装成功。

二、图像标注
首先获取应用场景下足够多的图像,根据需要将图像裁剪到固定尺寸,比如后期使用Unet模型训练,那么需要将图像裁剪为572*572,这些图像上包含了需要检测的各种类别。以遥感图像道路分割为例,这里收集了20张图像作为数据集。

启动labelme后,点击OpenDir打开数据集路径,通过点击create polygongs开始在图像上根据目标形状进行区域绘制,绘制完成后双击鼠标,在弹出窗口中输入类别名称。

完成所有图像的分割标注后,每张图像

本文详细介绍了如何使用labelme工具制作图像分割数据集,包括安装标注软件、图像标注、转换分割实例图像,并提供了相关脚本。最后,讨论了数据集的组织结构和后续训练模型的准备步骤。
最低0.47元/天 解锁文章
2259





