search

liuqk3 / PUT

Paper 'Transformer based Pluralistic Image Completion with Reduced Information Loss' in TPAMI 2024 and 'Reduce Information Loss in Transformers for Pluralistic Image Inpainting' in CVPR2022
MIT License
173 stars 15 forks source link

您好，如果使用自己的数据集的话，配置文件该如何修改呢 #22

Open zhangshilong2000 opened 11 months ago

liuqk3 commented 11 months ago

首先需要确认下现有ImageListDataset是否支持你的数据集，不合适的话需要自己写一个dataset，然后修改yaml文件中的dataloader就可以了