How to store .npy file as a folder containing images

Question

I am using the GRAD-CAM module that converts the images in a folder first to an array X and then after performing operations to a .npy file. But I would like to have this array back in image format and stored in a folder. How can I proceed further?

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')

score 0 · Accepted Answer · answered Jul 19 '21 at 13:01

0

If I understood right you want to save the output of gradcam as an image instead of .npy, right? In that case you can just replace your last line of code with something like cv2.imwrite("gradcam.png",gradcam), but you have to import cv2.

answered Jul 19 '21 at 13:01

Ramtin Nouri

310
1
8

But how do I do it for all the images in this long array X? – AMANDEEP KAUR Jul 19 '21 at 13:55
`for i in range(len(gradcam)): cv2.imwrite(F"{i}.png",gradcam[i])` ?? – Ramtin Nouri Jul 19 '21 at 13:57
1

Thanks! I tried, cv2.write but it saved a black image. Instead I used `for _ in range(len(gradcam)): matplotlib.image.imsave(F"/content/drive/MyDrive/Datasets/ircad/raw/grad-cam/{_}.png",gradcam[i]) ` – AMANDEEP KAUR Jul 19 '21 at 14:31

How to store .npy file as a folder containing images

1 Answers1