不想再熬叶 2022-09-28 14:05 采纳率: 60%
浏览 251

yolact训练报错

yolact训练自己的数据集报错,请问如何解决

loading annotations into memory...
Done (t=0.08s)
creating index...
index created!
loading annotations into memory...
Done (t=0.10s)
creating index...
index created!
C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\jit\_recursive.py:152: UserWarning: 'lat_layers' was found in ScriptModule constants,  but it is a non-constan
t submodule. Consider removing it.
  " but it is a non-constant {}. Consider removing it.".format(name, hint))
C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\jit\_recursive.py:152: UserWarning: 'pred_layers' was found in ScriptModule constants,  but it is a non-consta
nt submodule. Consider removing it.
  " but it is a non-constant {}. Consider removing it.".format(name, hint))
C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\jit\_recursive.py:152: UserWarning: 'downsample_layers' was found in ScriptModule constants,  but it is a non-
constant submodule. Consider removing it.
  " but it is a non-constant {}. Consider removing it.".format(name, hint))
Initializing weights...
Begin training!

Traceback (most recent call last):
  File "train.py", line 504, in <module>
    train()
  File "train.py", line 270, in train
    for datum in data_loader:
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 345, in __next__
    data = self._next_data()
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 856, in _next_data
    return self._process_data(data)
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 881, in _process_data
    data.reraise()
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\_utils.py", line 395, in reraise
    raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\worker.py", line 178, in _worker_loop
    data = fetcher.fetch(index)
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "D:\python-object\yolact-train\yolact\data\coco.py", line 94, in __getitem__
    im, gt, masks, h, w, num_crowds = self.pull_item(index)
  File "D:\python-object\yolact-train\yolact\data\coco.py", line 153, in pull_item
    target = self.target_transform(target, width, height)
  File "D:\python-object\yolact-train\yolact\data\coco.py", line 42, in __call__
    label_idx = self.label_map[label_idx] - 1
KeyError: 0
  • 写回答

3条回答 默认 最新

  • 我要吃木耳 2023-05-04 07:56
    关注

    楼主解决了嘛,一样的问题

    评论

报告相同问题?

问题事件

  • 修改了问题 9月28日
  • 创建了问题 9月28日