WARNING: FAILED to save image out/Gundam/validate/mask/val_000000_mask.png

[WakyWakyWayne]

print(result_dict)
{'shaded': tensor([[[1.0000, 1.0000, 1.0000],
[1.0000, 1.0000, 1.0000],
[1.0000, 1.0000, 1.0000],
...,
[1.0000, 1.0000, 1.0000],
[1.0000, 1.0000, 1.0000],
[1.0000, 1.0000, 1.0000]],

    [[1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     ...,
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000]],

    [[1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     ...,
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000]],

    ...,

    [[1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     ...,
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000]],

    [[1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     ...,
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000]],

    [[1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     ...,
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000],
     [1.0000, 1.0000, 1.0000]]], device='cuda:0'), 

mask': tensor([[[0.],
[0.],
[0.],
...,
[0.],
[0.],
[0.]],

    [[0.],
     [0.],
     [0.],
     ...,
     [0.],
     [0.],
     [0.]],

    [[0.],
     [0.],
     [0.],
     ...,
     [0.],
     [0.],
     [0.]],

    ...,

    [[0.],
     [0.],
     [0.],
     ...,
     [0.],
     [0.],
     [0.]],

    [[0.],
     [0.],
     [0.],
     ...,
     [0.],
     [0.],
     [0.]],

    [[0.],
     [0.],
     [0.],
     ...,
     [0.],
     [0.],
     [0.]]], device='cuda:0'), 

'normal': tensor([[[1., 1., 1.],
[1., 1., 1.],
[1., 1., 1.],
...,
[1., 1., 1.],
[1., 1., 1.],
[1., 1., 1.]],

    [[1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.],
     ...,
     [1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.]],

    [[1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.],
     ...,
     [1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.]],

    ...,

    [[1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.],
     ...,
     [1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.]],

    [[1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.],
     ...,
     [1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.]],

    [[1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.],
     ...,
     [1., 1., 1.],
     [1., 1., 1.],
     [1., 1., 1.]]], device='cuda:0')}

When
def save_image(fn, x : np.ndarray):
try:
if os.path.splitext(fn)[1] == ".png":
imageio.imwrite(fn, np.clip(np.rint(x * 255.0), 0, 255).astype(np.uint8), compress_level=3) # Low compression for faster saving
else:
imageio.imwrite(fn, np.clip(np.rint(x * 255.0), 0, 255).astype(np.uint8))
except:

    print("WARNING: FAILED to save image %s" % fn)

I cant save mask.png. And the mask shape is (512,512,1), it looks no issue.
normal and shaded PNG are OK,only mask failed.

[WakyWakyWayne]

OK,I find the problem.
imageio cant save single channel mask.png, here need cv2 instead

[ruiqixu37]

OK,I find the problem. imageio cant save single channel mask.png, here need cv2 instead

This solved my problem too. Thanks!