fnames = list(Path("./segment_huskies/img/").glob("*")) fn = fnames[0] def load_img_msk(fn): return PILImage.create(fn), PILMask.create(fn.parent.parent / "msk" / fn.name) img, msk = load_img_msk(fn) show_images([img,msk])