Using VOC format dataset can be trained normally, conversion to COCO format error... Other items of the dataset can be used normally.
please help me……
2022-05-15 20:14:15,830 SSD.trainer INFO: Start training ...
Traceback (most recent call last):
File "/home/lhaozz/hand_ssd/train.py", line 118, in
main()
File "/home/lhaozz/hand_ssd/train.py", line 109, in main
model = train(cfg, args)
File "/home/lhaozz/hand_ssd/train.py", line 47, in train
model = do_train(cfg, model, train_loader, optimizer, scheduler, checkpointer, device, arguments, args)
File "/home/lhaozz/hand_ssd/ssd/engine/trainer.py", line 74, in dotrain
for iteration, (images, targets, ) in enumerate(data_loader, start_iter):
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/dataset.py", line 308, in getitem
return self.datasets[dataset_idx][sample_idx]
File "/home/lhaozz/hand_ssd/ssd/data/datasets/coco.py", line 51, in getitem
image = self._read_image(image_id)
File "/home/lhaozz/hand_ssd/ssd/data/datasets/coco.py", line 92, in _read_image
file_name = self.coco.loadImgs(image_id)[0]['file_name']
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/pycocotools/coco.py", line 227, in loadImgs
return [self.imgs[id] for id in ids]
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/pycocotools/coco.py", line 227, in
return [self.imgs[id] for id in ids]
KeyError: '6'
Using VOC format dataset can be trained normally, conversion to COCO format error... Other items of the dataset can be used normally. please help me……
2022-05-15 20:14:15,830 SSD.trainer INFO: Start training ... Traceback (most recent call last): File "/home/lhaozz/hand_ssd/train.py", line 118, in
main()
File "/home/lhaozz/hand_ssd/train.py", line 109, in main
model = train(cfg, args)
File "/home/lhaozz/hand_ssd/train.py", line 47, in train
model = do_train(cfg, model, train_loader, optimizer, scheduler, checkpointer, device, arguments, args)
File "/home/lhaozz/hand_ssd/ssd/engine/trainer.py", line 74, in dotrain
for iteration, (images, targets, ) in enumerate(data_loader, start_iter):
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/torch/utils/data/dataset.py", line 308, in getitem
return self.datasets[dataset_idx][sample_idx]
File "/home/lhaozz/hand_ssd/ssd/data/datasets/coco.py", line 51, in getitem
image = self._read_image(image_id)
File "/home/lhaozz/hand_ssd/ssd/data/datasets/coco.py", line 92, in _read_image
file_name = self.coco.loadImgs(image_id)[0]['file_name']
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/pycocotools/coco.py", line 227, in loadImgs
return [self.imgs[id] for id in ids]
File "/home/lb511/anaconda3/envs/Lhaopy/lib/python3.9/site-packages/pycocotools/coco.py", line 227, in
return [self.imgs[id] for id in ids]
KeyError: '6'
i have 10 categories..