ssd.pytorch
ssd.pytorch copied to clipboard
KeyError: 'Traceback (most recent call last):\n File "E:\\anaconda\\anaconda\\envs\\pytorch\\lib\\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 "E:\\anaconda\\anaconda\\envs\\pytorch\\lib\\site-packages\\torch\\utils\\data\\dataloader.py", line 138, in <listcomp>\n samples = collate_fn([dataset[i] for i in batch_indices])\n File "F:\\ssd.pytorch-master\\data\\voc0712.py", line 111, in __getitem__\n im, gt, h, w = self.pull_item(index)\n File "F:\\ssd.pytorch-master\\data\\voc0712.py", line 126, in pull_item\n target = self.target_transform(target, width, height)\n File "F:\\ssd.pytorch-master\\data\\voc0712.py", line 68, in __call__\n label_idx = self.class_to_ind[name]\nKeyError: \'pointer_instrument\'\n'
KeyError: 'Traceback (most recent call last):\n File "E:\anaconda\anaconda\envs\pytorch\lib\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 "E:\anaconda\anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 138, in
I have meet the same error have you solve it? thanks.
same error. Have you guys solve it?
same issue, have you solved it ? thanks a lot
Loading base network...
Initializing weights...
Loading the dataset...
Training SSD on: SMOKE
Using the specified args:
Namespace(basenet='vgg16_reducedfc.pth', batch_size=8, cuda=True, dataset='VOC', dataset_root='E:/ssd.pytorch/data/VOCdevkit/VOCsmoke/', gamma=0.1, lr=1e-05, momentum=0.9, num_workers=4, resume=None, s4, resume=None, save_folder='E:/ssd.pytorch/weights/', start_iter=0, visdom=False, weight_decay=0.0005)
Traceback (most recent call last):
File ".\train.py", line 254, in
and the line 156 is : try: images, targets = next(batch_iterator) except StopIteration: batch_iterator = iter(data_loader) images, targets = next(batch_iterator)