image_output_1.png
working_path = Path.cwd()
folders = ('train', 'test')
labels = ('0', '1')

input_path = Path('/kaggle/input')
train_image_paths = sorted(input_path.rglob('train/*.png'))
test_image_paths = sorted(input_path.rglob('test/*.png'))

train_image_paths