将图像转换为npz文件

1,将图像转为npz文件进行存储

import os
from PIL import Image
import numpy as np
path = "INbreast/CC_image/CCtestresize/"
files = os.listdir(path)
list_img_3d = []
for file in files:
    if "DS_Store" in file:
        continue
    img = Image.open(path+file).convert("L")
    list_img_3d.append(np.array(img))
arr_img_3d = np.array(list_img_3d)
np.savez(os.path.join("img_dir", "test_b9.npz"), vol = arr_img_3d)

2,可视化npz文件

import numpy as np
import os
data_atlas = np.load("img_dir/atlas_norm_2d.npz")
r = data_atlas["vol"].T
plt.imshow(r)
plt.show()
发布了315 篇原创文章 · 获赞 119 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/w144215160044/article/details/104915734