如何为该细分网络制作好的遮罩文件(如原始数据集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中打开了所有黑色文件。
我用phoshop制作遮罩文件-背景黑色(255,255,255),人形-白色(0、0、0)。 但是用我的文件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 ...