train failure: IndexError: tuple index out of range

[ZhengSJCode]

我该如何处理这个问题

用vgg 1shot 在split0训练是可以的，但是在split1和split2 出现以下问题，好像是因为标签全是background

Traceback (most recent call last):
File "train.py", line 560, in
main()
File "train.py", line 217, in main
train(train_loader, val_loader, model, optimizer, epoch, scaler)
File "train.py", line 291, in train
for i, (input, target, s_input, s_mask, subcls) in enumerate(train_loader):
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 363, in next
data = self._next_data()
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 971, in _next_data
return self._process_data(data)
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1014, in _process_data
data.reraise()
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/_utils.py", line 395, in reraise
raise self.exc_type(msg)
IndexError: Caught IndexError in DataLoader worker process 6.
Original Traceback (most recent call last):
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 185, in _worker_loop
data = fetcher.fetch(index)
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/ustc/miniconda3/envs/fsdet/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/ustc/repository/R2Net/dataset/base_data.py", line 235, in getitem
support_label = label_trans(support_label, class_chosen, mode='To_zero')
File "/home/ustc/repository/R2Net/dataset/base_data.py", line 27, in label_trans
new_label[target_pix[0],target_pix[1]] = 1
IndexError: tuple index out of range

[Binfeitu]

，有没有更详细的描述，你说在split1/2时，均是背景，请问是不是图片本身有问题，你可以在imread后再看看