rtdetrv2更换数据集后出现报错，请问应该怎么解决 #592
Description
File "/root/data1/RT-DETR-main/rtdetrv2_pytorch/tools/../src/solver/det_solver.py", line 38, in fit
train_stats = train_one_epoch(
File "/root/data1/RT-DETR-main/rtdetrv2_pytorch/tools/../src/solver/det_engine.py", line 38, in train_one_epoch
for i, (samples, targets) in enumerate(metric_logger.log_every(data_loader, print_freq, header)):
File "/root/data1/RT-DETR-main/rtdetrv2_pytorch/tools/../src/misc/logger.py", line 215, in log_every
for obj in iterable:
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 708, in next
data = self._next_data()
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1480, in _next_data
return self._process_data(data)
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1505, in _process_data
data.reraise()
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/_utils.py", line 733, in reraise
raise exception
NotImplementedError: Caught NotImplementedError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/utils/data/_utils/worker.py", line 349, in _worker_loop
data = fetcher.fetch(index) # type: ignore[possibly-undefined]
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 52, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/data1/RT-DETR-main/rtdetrv2_pytorch/tools/../src/data/dataset/coco_dataset.py", line 41, in getitem
img, target, _ = self._transforms(img, target, self)
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
File "/root/data1/RT-DETR-main/rtdetrv2_pytorch/tools/../src/data/transforms/container.py", line 46, in forward
return self.get_forward(self.policy['name'])(*inputs)
File "/root/data1/RT-DETR-main/rtdetrv2_pytorch/tools/../src/data/transforms/container.py", line 74, in stop_epoch_forward
sample = transform(sample)
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torchvision/transforms/v2/_transform.py", line 68, in forward
flat_outputs = [
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torchvision/transforms/v2/_transform.py", line 69, in
self.transform(inpt, params) if needs_transform else inpt
File "/root/miniconda3/envs/rtdetrv2/lib/python3.10/site-packages/torchvision/transforms/v2/_transform.py", line 55, in transform
raise NotImplementedError
NotImplementedError