import cv2 import numpy as np matrix = np.reshape(data_train_batch1[0], (3, 32, 32)) matrix = matrix.transpose(1, 2, 0) cv2.imwrite("img_test_show.png", matrix) img = cv2.imread("img_test_show.png") cv2.imshow('img', img) cv2.waitKey(0) cv2.destroyAllWindows()