Dataset is empty if configuring img_suffix: .jpg in default.yaml

[linmingren]

When I place the files "test.jpg" and "test_mask001.jpg" in the "indir" directory and then run "predict.py", no images are generated in the output directory because the dataset is empty.

I found the suffix of mask_filenames in saicinpainting\evaluation\data.py is hard-coded to ".png":
class InpaintingDataset(Dataset): def __init__(self, datadir, img_suffix='.jpg', pad_out_to_modulo=None, scale_factor=None): self.datadir = datadir self.mask_filenames = sorted(list(glob.glob(os.path.join(self.datadir, '**', '*mask*.png'), recursive=True))) self.img_filenames = [fname.rsplit('_mask', 1)[0] + img_suffix for fname in self.mask_filenames] self.pad_out_to_modulo = pad_out_to_modulo self.scale_factor = scale_factor
I think it should be 'mask' + img_suffix instead of 'mask.png'.