今天剛發現對多維numpy數組使用random.shuffle存在一些問題
# 把mask為0的值改為2 tmp_mask = (mask == 0).astype(np.int64) * 2 + mask # 把tmp_mask上對應image值為-32767的點置為0 tmp_mask = np.where(tmp_image != -32767, tmp_mask, np.zeros_like(tmp_mask)) # 邊緣切割1個像素使圖像為整數 image = image[:, :1200, :1200, :] mask = mask[:, :1200, :1200] tmp_image = tmp_image[:, :1200, :1200] tmp_mask = tmp_mask[:, :1200, :1200] random.seed(628) random.shuffle(image) random.seed(628) random.shuffle(mask) random.seed(628) random.shuffle(tmp_image) random.seed(628) random.shuffle(tmp_mask) random.seed(628) random.shuffle(filenames_array)
在random.shuffle之前
在random.shuffle之后
可見random.shuffle直接作用於多維numpy數組並不會只打亂第一維數據（具體怎么打亂目前尚不清楚，請高人指點）