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

[ssbilakeri]

"__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]

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.