如何为分割网制作新的图像数据集

时间:2019-08-14 16:19:08

标签: tensorflow dataset

如何为该细分网络制作好的遮罩文件(如原始数据集oxford_iiit_pet:3.0.0)?我想使用photoshop,可以通过numpy或matlab进行转换

我正在学习张量流分割-示例https://www.tensorflow.org/beta/tutorials/images/segmentation。我运行thins示例,它可以工作。我想为其制作特殊的数据集UNET。数据集必须类似于oxford_iiit_pet:3.0.0。问题是:在原始数据集蒙版中制作png文件批注/trimaps/...png。我在phoshop中打开了所有黑色文件。

enter image description here我用phoshop制作遮罩文件-背景黑色(255,255,255),人形-白色(0、0、0)。 enter image description here 但是用我的文件UNET无法工作,有错误。问题是遮罩文件格式。

我的代码与示例https://www.tensorflow.org/beta/tutorials/images/segmentation相同,仅其他数据集。型号是:

base_model = tf.keras.applications.MobileNetV2(input_shape=[128, 128, 3], include_top=False)

有错误: Invalid argument: Received a label value of -1 which is outside the valid range of [0, 3). Label values: -1 -1 -1 -1 -1 -1 -1 -1 ...

0 个答案:

没有答案