-
Notifications
You must be signed in to change notification settings - Fork 7
KeyError: 'edges_negative' #3
Description
Traceback (most recent call last):
File "/home/pete/anaconda3/envs/mamba/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/pete/anaconda3/envs/mamba/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/home/pete/space/PLNet-master/hawp/fsl/train.py", line 271, in
train(cfg, model, train_dataset, optimizer, scheduler, loss_reducer, checkpointer, arguments)
File "/home/pete/space/PLNet-master/hawp/fsl/train.py", line 112, in train
for it, (images, annotations) in enumerate(train_dataset):
File "/home/pete/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 630, in next
data = self._next_data()
File "/home/pete/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 674, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/home/pete/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/pete/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/pete/space/PLNet-master/hawp/fsl/dataset/train_dataset.py", line 201, in getitem
if len(ann['edges_negative']) == 0:
KeyError: 'edges_negative'