运行test.py出错

[wangluyao623]

data data.reraise() File "/home/lsl/anaconda3/envs/wlyy/lib/python3.8/site-packages/torch/_utils.py", line 644, in reraise raise exception TypeError: Caught TypeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/lsl/anaconda3/envs/wlyy/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/home/lsl/anaconda3/envs/wlyy/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/lsl/anaconda3/envs/wlyy/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/data1/lsl/wly/CDRL/datasets.py", line 54, in getitem transformed_A = self.transforms(image=img_A) File "/home/lsl/anaconda3/envs/wlyy/lib/python3.8/site-packages/albumentations/core/composition.py", line 195, in call self._check_args(**data) File "/home/lsl/anaconda3/envs/wlyy/lib/python3.8/site-packages/albumentations/core/composition.py", line 275, in _check_args raise TypeError("{} must be numpy array type".format(data_name)) TypeError: image must be numpy array type 请问应该如何解决