错误描述:
KeyError: 'Traceback (most recent call last):\n File "/home/em/anaconda3/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop\n samples = collate_fn([dataset[i] for i in batch_indices])\n File "/home/em/anaconda3/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 138, in <listcomp>\n samples = collate_fn([dataset[i] for i in batch_indices])\n File "/home/data_1/project/big-obj/CenterNet/src/lib/datasets/sample/ctdet.py", line 102, in __getitem__\n cls_id = int(self.cat_ids[ann[\'category_id\']])\nKeyError: 12\n'
原因:直接使用了coco.py修改自己的数据集,coco.py中少了12.
无语..........................................................