KeyError: тензор (1) при попытке обучить модель Faster R-CNN ⇐ Python

[Anonymous]

Я получаю следующую ошибку при попытке обучить мою модель Faster R-CNN. Может ли кто-нибудь отследить источник ошибки или узнать, в чем дело?:
Код для обучения:
!python train. py --data data_configs/custom_data.yaml --epochs 1 --imgsz 850 --model fastrcnn_resnet50_fpn_v2 --name custom_training --batch 4 --device cuda
Ошибка:
Файл «C:\Users..\fastercnn-pytorch-training\train.py», строка 574, в
main(args)
Файл «C:\Users..\fastercnn-pytorch-training\fastercnn-pytorch-training\train.py», строка 414, в основном пакете_loss_rpn_list = train_one_epoch(
Файл «C» :\Users..\fastercnn-pytorch-training\fastercnn-pytorch-training\torch_utils\engine.py", строка 46, в train_one_epoch
для изображений, цели в metric_logger.log_every(data_loader, print_freq, header):
Файл «C:\Users..\fastercnn-pytorch-training\fastercnn-pytorch-training\torch_utils\utils.py», строка 174, в log_every
для пакета в итерируемом:
Файл "C:\Users..\AppData\Roaming\Python\Python310\site-packages\torch\utils\data\dataloader.py", строка 701, в следующий
data = self._next_data()
Файл "C:\Users..\AppData\Roaming\Python\Python310\site-packages" \torch\utils\data\dataloader.py", строка 757, в _next_data
data = self._dataset_fetcher.fetch(index) # может вызвать StopIteration
Файл "C: \Users..\AppData\Roaming\Python\Python310\site-packages\torch\utils\data_utils\fetch.py", строка 52, в fetch
data = [self.dataset[idx] для idx в возможно_пакетном_индексе ]
Файл «C:\Users..\AppData\Roaming\Python\Python310\site-packages\torch\utils\data_utils\fetch.py», строка 52, в
data = [self.dataset[idx] для idx в возможно_batched_index]
Файл "C:\Users..\fastercnn-pytorch-training\fastercnn-pytorch-training\datasets. py", строка 317, в getitem
sample = self.transforms(image=image_resized,
файл "C:\Users..\AppData \Roaming\Python\Python310\site-packages\albumentations\core\composition.py", строка 421, в call
self.preprocess(data)
Файл «C:\Users..\AppData\Roaming\Python\Python310\site-packages\albumentations\core\composition.py», строка 447, в предварительной обработке
p.preprocess(data)
Файл «C:\Users..\AppData\Roaming\Python\Python310\site-packages\albumentations\core\utils.py», строка 171, в предварительной обработке
data = self.add_label_fields_to_data(data)
Файл "C:\Users..\AppData\Roaming\Python\Python310\site-packages\albumentations\core\utils.py", строка 220, в add_label_fields_to_data
data[data_name] = self._process_label_fields(data, data_name)
Файл "C:\Users..\AppData\Roaming\Python\Python310\site-packages" \albumentations\core\utils.py", строка 229, в _process_label_fields
encoded_labels = self._encode_label_field(data, data_name, label_field)
Файл "C:\Users.. \AppData\Roaming\Python\Python310\site-packages\albumentations\core\utils.py", строка 249, в _encode_label_field
encoded_labels = encoder.fit_transform(data[label_field]).reshape(-1, 1)
Файл "C:\Users...\AppData\Roaming\Python\Python310\site-packages\albumentations\core\utils.py", строка 84, в fit_transform
вернуть self.transform(y)
Файл "C:\Users...\AppData\Roaming\Python\Python310\site-packages\albumentations\core\utils.py", строка 80, в преобразовании
return np.array([self.classes_[label] для метки в y])
File "C:\Users...\AppData\ Roaming\Python\Python310\site-packages\albumentations\core\utils.py", строка 80, в
return np.array([self.classes_[label] for label in y])KeyError: tensor(1)

Подробнее здесь: https://stackoverflow.com/questions/791 ... -cnn-model