使用labelme工具标注人像数据集
一、前言
在《基于人脸识别和图像分割技术制作证件照》博文中讲解了如何基于分离出的人像制作证件照,在《基于PASCAL VOC 2012数据集训练deeplab图像分割模型》博文中讲解了如何训练图像分割模型,但是PASCAL VOC 2012数据集不是专注于人像分割任务的数据,因此需要借助标注工具标注人像分割数据进一步调优deeplab模型。今天教大家使用labelme工具标注专注于人像分割任务的数据。
二、安装labelme工具
- 创建并激活labelme python环境
conda安装与使用请参考基于anaconda管理python包版本
conda create -n labelme python=3.6
conda activate labelme
- 安装依赖
pip install pyqt5
- 安装labelme
pip install labelme
三、使用
其中2~4步请观看视频,使用lablelme标注人像分割数据
- 启动labelme
lableme
- 启用相关窗口
- 打开图片目录,安如图所示操作
- 标注人像,按如图所示标注,可参考视频
- 选择下一张图片继续标注
- 转为PASCAL VOC格式数据
git clone https://github.com/wkentaro/labelme.git
cd labelme/examples/semantic_segmentation/
./labelme2voc.py data_annotated data_dataset_voc --labels labels.txt # data_annotated为第4步保存的json数据所在路径
转换后将产生如下的路径
# - data_dataset_voc/JPEGImages
# - data_dataset_voc/SegmentationClass
# - data_dataset_voc/SegmentationClassVisualization
todo
- 下载中国人肖像数据,已完成,请参考《从视觉中国网站爬取中国人肖像(人像)》
- 基于自定义数据集调优deeplab模型