如何將 .npy 文件存儲為包含圖像的文件夾

Question

我正在使用 GRAD-CAM 模塊，該模塊首先將文件夾中的圖像轉換為數組 X，然后在執行操作后轉換為 .npy 文件。 但我想讓這個數組恢復為圖像格式並存儲在一個文件夾中。 我怎樣才能進一步？

dir_path = '/content/drive/MyDrive/Datasets/ircad/raw/Original_sliced_train/'
files = os.listdir(dir_path)
N = len(files)

X = np.empty((N, H, W, 3))
for i, file in enumerate(files):
     x = image.load_img(dir_path + file, target_size=(H, W))
     X[i] = image.img_to_array(x)
X = preprocess_input(X)

---

top = np.argmax(model.predict(X), 1)

---

gradcam = np.empty((X.shape[:-1]))
batch_size = 32
for i in range((N + batch_size - 1) // batch_size):
    start = i * batch_size
    end = min((i+1) * batch_size, N)
    gradcam[start:end] = grad_cam_batch(model, X[start:end], top[start:end], 'block5_conv3')

gradcam.tofile('gradcam.npy')

Answer 1

如果我理解正確，您想將 gradcam 的輸出保存為圖像而不是 .npy，對嗎？ 在這種情況下，您可以將最后一行代碼替換為cv2.imwrite("gradcam.png",gradcam) ，但您必須import cv2 。