Event-AHU / OpenPAR

[OpenPAR] An open-source framework for Pedestrian Attribute Recognition, based on PyTorch
MIT License
79 stars 11 forks source link

关于生成Pad_datasets ,是需要自己实现吗? #3

Closed yujun686 closed 9 months ago

yujun686 commented 10 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的情况

yujun686 commented 10 months ago

上述问题在将peat_pad中dataset.root = os.path.join(save_dir, 'Pad_datasets')修改为dataset.root = os.path.join(save_dir, 'images')后得到解决,但修改后似乎和pata.py 内容是一样的

1125178969 commented 10 months ago

Pad_datasets需要自己对数据集进行处理 ,使用黑色像素填充为H=W即可