Как сохранить изображения, используя для цикла? - PullRequest
Новая
       15

Как сохранить изображения, используя для цикла?

0 голосов
/ 02 мая 2019

У меня есть набор данных для проблемы сегментации.Данные хранятся в расширении .h5.Каждая точка данных т.е.«1234.h5» включает изображение, маску и некоторую другую информацию об изображении.То, что я хотел бы сделать, это извлечь изображения и их маски и назвать их image_0, image_1, ... mask_0, mask_1, ... Попытка написать цикл для этого, но не удалось.

Мне удалось извлечь изображения и маски по одному.Но не в цикле. 

imagenames = pd.read_hdf("imagenames.h5")
image_paths = list(imagenames["annotated_image_path"])

# Read in the image and the mask
def get_visible_and_mask(image_path):
    with h5py.File(image_path, 'r') as annotated_img:
        img_visible = annotated_img["georef_img"]["layers"]["visible"]["array"][()]
        mask = annotated_img["georef_img"]["layers"]["mask"]["array"][()]
    return img_visible, mask

# Get a random image_path
image_path = random.choice(image_paths)
img_visible, mask = get_visible_and_mask(image_path)

#Save the random image and the mask
imageio.imwrite('img_vis.tif', img_visible)
imageio.imwrite('mask.tif', mask)
...