Cannot handle this data type

[LYL-hub-source]

@avalonstrel
Excuse me, when I run my own data set, the following error occurs, how should I solve it?
File "/home/lyl/下载/anaconda3/envs/torch/lib/python3.6/site-packages/PIL/Image.py", line 2837, in fromarray
raise TypeError("Cannot handle this data type: %s, %s" % typekey) from e
TypeError: Cannot handle this data type: (1, 1, 1, 9), |u1

[jackhu-bme]

Same strange error now. I wonder how In could change the Image.fromarray function. I still meet this error even when I change the numpy array to np.unint8 type and check the correct shape. I am really confused about this.

[jackhu-bme]

I want to use this repo for baseline comparison, but the error is reallt confusing.

[StrideZH]

shape of mask from .pkl must be (height, width, 1)