text_plain_output_1.png
import numpy as np

import os

import os

train_images_dir = 'train_images'
train_maps_dir = 'train_labels'
val_images_dir = 'valid_images'
val_maps_dir = 'valid_labels'
test_images_dir = 'test_images'
test_maps_dir = 'test_labels'
train_images = np.array(os.listdir(train_images_dir), dtype=object)
train_maps = np.array(os.listdir(train_maps_dir), dtype=object)
print('****TRAIN****')
print(f'There are {len(train_images)} images')
print(f'There are {len(train_maps)} masks')
val_images = np.array(os.listdir(val_images_dir), dtype=object)
val_maps = np.array(os.listdir(val_maps_dir), dtype=object)
print('****VALID****')
print(f'There are {len(val_images)} images')
print(f'There are {len(val_maps)} masks')