Con endswith()
es más sencillo el código en el filtrado de los archivos mask:
dataset_path = './'
filenames = os.listdir(dataset_path)
filenames.sort()
image_list = []
mask_list = []
for filename in filenames:
if len(filename.split('.')) > 1 and filename.endswith('png'):
if filename.split('.')[0].endswith('L'):
mask_list.append(dataset_path + filename)
else:
image_list.append(dataset_path + filename)
¿Quieres ver más aportes, preguntas y respuestas de la comunidad?