huanghoujing / EANet

EANet: Enhancing Alignment for Cross-Domain Person Re-identification
397 stars 88 forks source link

KeyError: 'Market-1501-v15.09.15/bounding_box_train\\0661_c5s2_040055_01.jpg' #34

Open ssbilakeri opened 3 years ago

ssbilakeri commented 3 years ago
"__main__", mod_spec)

File "C:\Users\padmashree\anaconda3\envs\savitri\lib\runpy.py", line 85, in _run_code exec(code, run_globals) File "D:\project_dir\EANet\package\optim\eanet_trainer.py", line 135, in trainer.train_phases() File "D:\project_dir\EANet\package\optim\eanet_trainer.py", line 126, in train_phases self.train() File "D:\project_dir\EANet\package\optim\reid_trainer.py", line 338, in train self.trainer.train_one_epoch(trial_run_steps=3 if cfg.trial_run else None) File "D:\project_dir\EANet\package\optim\trainer.py", line 35, in train_one_epoch for i, batch in enumerate(self.train_loader): File "C:\Users\padmashree\anaconda3\envs\savitri\lib\site-packages\torch\utils\data\dataloader.py", line 615, in next batch = self.collate_fn([self.dataset[i] for i in indices]) File "C:\Users\padmashree\anaconda3\envs\savitri\lib\site-packages\torch\utils\data\dataloader.py", line 615, in batch = self.collate_fn([self.dataset[i] for i in indices]) File "D:\project_dir\EANet\package\data\dataset.py", line 41, in getitem sample['pap_mask'] = self.get_pap_mask(im_path) File "D:\project_dir\EANet\package\data\dataset.py", line 74, in get_pap_mask kpt = self.im_path_to_kpt[key]['kpt'] KeyError: 'Market-1501-v15.09.15/bounding_box_train\0661_c5s2_040055_01.jpg'

help me to resolve this

huanghoujing commented 3 years ago

Hi, Market-1501-v15.09.15/bounding_box_train\0661_c5s2_040055_01.jpg is not a valid key given my provided files. You may have to try Market-1501-v15.09.15/bounding_box_train/0661_c5s2_040055_01.jpg.