Error in utils.build_super_images

[ValerioNeriGit]

Hi,

when running with the command python main.py --cfg cfg/bird_cycle.yaml --gpu 0 the function crashes on PIL_att = Image.fromarray(np.uint8(one_map)).

This is the error:

 File "/Data/anaconda3/envs/cyclegan-py3/lib/python3.6/site-packages/PIL/Image.py", line 2772, in fromarray
    mode, rawmode = _fromarray_typemap[typekey]
KeyError: ((1, 1, 6), '|u1')

I guess is a problem of dimensions, one_map ha dimension (128, 128, 6) while img has (128, 128, 3).

Any ideas how to solve it?
Thanks

[ScottWong98]

In utils.py, change the usage of skimage.transform.pyramid_expand function: add multichannel=True when use this function. In this way, the last dimension will not be changed.
https://scikit-image.org/docs/dev/api/skimage.transform.html?highlight=skimage%20transform%20pyramid_expand#skimage.transform.pyramid_expand