制作分割数据集,数据集标完,想扩充怎么办,图片扩充方便,但是标签怎么扩充?
本文针对3分类问题,是个笨方法:背景,目标1,目标2。
数据集类型:voc数据集
格式:图片(.jpg),标签(.png)
程序获取:文章最后有百度网盘链接
首先,利用数据增广文件,对图片和标签进行旋转,镜像等。
然后,发现问题解决问题。或者,针对最开始,数据增广时,就思考,如何增广出正确的标签。
对标签进行检查:检查是否为单通道(8位深),检查像素点是否是[0,n-1],其中n代表voc数据集中的类,同时voc数据集类的个数是背景+目标数,本文中n为3,标签像素值应为0,1,2。
问题1:扩展后的标签是3通道(24位深),但要求是1通道(8位深)。
用多通道改单通道程序,将其改为1通道,百度网盘有
问题2:标签像素值有误,
一张正确标签像素值为
错误标签值为
这种像素值的图像在深度网络中是跑不通的。
如何修改,这就讲到笨方法了。通过检查像素值程序,对原始标签图片和变换后的标签图片进行检查,最好是该原始图片中只有一个分割目标标签,伪彩图分割颜色全红或者全绿。我的数据中扩展中的12,13像素值,代表的是1,另外22,28代表的是2。确认后用转标签程序Convert_SegmentationClass.py,对其进行修改,只需要改Origin_Point_Value和Out_Point_Value,一一对应关系即可
再次检验修改后的标签图片,可以发现像素值是理想中的0,1,2了。这时候图像变成全黑,是没问题的,可以正常使用。可以发现异常扩充标签图像的12,13的像素值之和,为1的像素值数,即value栏。
注意,在使用过程中,做多只需要改路径,模块都是通用的,最多缺什么模块pip install 模块。填入路径后,有时候文件夹最后面需要加斜杠,才能正确读取到文件。如:
D:\xlypostgraduatestudy\2codedownload--------->D:/xlypostgraduatestudy/2codedownload/
链接:https://pan.baidu.com/s/1FO70odWXcZkPyajZoj060w
提取码:1234
--来自百度网盘超级会员V1的分享