TypeError: Cannot handle this data type: (1, 1, 6), |u1
阿新 • • 發佈：2021-10-15
PIL_im=Image.fromarray(np.uint8(img))
PIL_att=Image.fromarray(np.uint8(one_map)).convert('RGB')
執行一部分之後報錯
TypeError: Cannot handle this data type: (1, 1, 6), |u1
第一個解決方案
PIL需要的格式是（W,H,C），而資料集的格式是（C,W,H）所以要進行轉換，把（C,W,H）變為（W,H,C）
把最上邊兩行程式碼改成
PIL_im=Image.fromarray(np.uint8(img.transpose(1,2,0)))
PIL_att=Image.fromarray(np.uint8(one_map.transpose(1,2,0))).convert('RGB')
from
length_array_to_mask_tensor()table_dict_to_mask_tensor()/pred_col_mask()from
mask = torch.ByteTensor(mask)tomask = torch.BoolTensor(mask).