Closed yujun686 closed 9 months ago
在执行完 python dataset/preprocess/peta_pad.py 后,仅生成了pad.pkl并未产生Pad_datasets, 执行python train.py PETA --use_text_prompt --use_div --use_vismask --use_GL --use_mm_former 会出现找不到/PETA/Pad_datasets/10807.png的情况
上述问题在将peat_pad中dataset.root = os.path.join(save_dir, 'Pad_datasets')修改为dataset.root = os.path.join(save_dir, 'images')后得到解决,但修改后似乎和pata.py 内容是一样的
Pad_datasets需要自己对数据集进行处理 ,使用黑色像素填充为H=W即可
在执行完 python dataset/preprocess/peta_pad.py 后,仅生成了pad.pkl并未产生Pad_datasets, 执行python train.py PETA --use_text_prompt --use_div --use_vismask --use_GL --use_mm_former 会出现找不到/PETA/Pad_datasets/10807.png的情况