ZhengyuLiang24 / LFT

PyTorch Implementation of "Light Field Image Super-Resolution with Transformers"
MIT License
86 stars 14 forks source link

IndexError: too many indices for array: array is 0-dimensional, but 2 were indexed #13

Open ZQ-perfect opened 1 year ago

ZQ-perfect commented 1 year ago

你好,请问一下我用你给的数据训练时进行trainsetdataloader出现下述问题: File "C:\Users\Lenovo\anaconda3\envs\PY\lib\site-packages\torch\utils\data_utils\fetch.py", line 58, in data = [self.dataset[idx] for idx in possibly_batched_index] File "C:\Users\Lenovo\PycharmProjects\pythonProject1\LFT\utils\utils_datasets.py", line 40, in getitem data_SAI_y, label_SAI_y = augmentation(data_SAI_y, label_SAI_y) File "C:\Users\Lenovo\PycharmProjects\pythonProject1\LFT\utils\utils_datasets.py", line 116, in augmentation data = data[:, ::-1] IndexError: too many indices for array: array is 0-dimensional, but 2 were indexed 请问是数据问题吗