liyuke65535 / Part-Aware-Transformer

[ICCV 2023] An official implementation for "Part-Aware Transformer for Generalizable Person Re-identification"
29 stars 3 forks source link

veriwild.py erro #8

Open yangjunxiu opened 10 months ago

yangjunxiu commented 10 months ago

erro info: Traceback (most recent call last): File "E:\Software\pytorch\pycharm\PyCharm 2023.1.2\plugins\python\helpers\pydev\pydevd.py", line 1496, in _exec pydev_imports.execfile(file, globals, locals) # execute the script ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "E:\Software\pytorch\pycharm\PyCharm 2023.1.2\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "F:\occlusion\Part-Aware-Transformer\train.py", line 71, in train_loader = build_reid_train_loader(cfg) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\occlusion\Part-Aware-Transformer\data\build_DG_dataloader.py", line 42, in build_reid_train_loader dataset = DATASET_REGISTRY.get(d)(root=_root, combineall=cfg.DATASETS.COMBINEALL) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\occlusion\Part-Aware-Transformer\data\datasets\veriwild.py", line 47, in init train = self.process_dir(self.train_list) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\occlusion\Part-Aware-Transformer\data\datasets\veriwild.py", line 64, in process_dir dataset.append((self.imgid2imgpath[imgid], vid, int(self.imgid2camid[imgid])))


KeyError: '119083.jpg 0 23'
python-BaseException

hi, Sr, have you ever tried the VeRi-wild datasets on your code? it seems some errors there.
liyuke65535 commented 2 months ago

Sorry about that. Since we just consider person ReID, I have never tried VeRi-wild. But based on the info you give, you are recommended to debug and check the whole data preprocess.