IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed

[afefbnsaid]

hi,
I tried to train the SMSR with my data but i got this error.
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed
can you help me to solve it :)

[LongguangWang]

Hi @afefbnsaid, thanks for you interest in our work. Please provide more details about your training data and the error information (e.g., which line in which file) such that we can better help you.

[afefbnsaid]

My data were divided into two categories: super resolution (visible images) and low resolution (thermal images).
#error details
700
/content/drive/MyDrive/database/benchmark/Set5/HR
/content/drive/MyDrive/database/benchmark/Set5/LR_bicubic
0
Making model...
Preparing loss function:
1.000 * L1
[Epoch 1] Learning rate: 2.00e-4
Traceback (most recent call last):
File "main.py", line 20, in
t.train()
File "/content/drive/My Drive/SMSR/trainer.py", line 43, in train
for batch, (lr, hr, _, idx_scale) in enumerate(self.loader_train):
File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
IndexError: Traceback (most recent call last):
File "/content/drive/My Drive/SMSR/dataloader.py", line 38, in _ms_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/content/drive/My Drive/SMSR/dataloader.py", line 38, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/content/drive/My Drive/SMSR/data/multiscalesrdata.py", line 144, in getitem
lr, hr = self.get_patch(lr, hr)
File "/content/drive/My Drive/SMSR/data/multiscalesrdata.py", line 195, in get_patch
multi_scale=multi_scale
File "/content/drive/My Drive/SMSR/data/common.py", line 35, in get_patch
args[0][iy:iy + ip, ix:ix + ip, :],*[a[ty:ty + tp, tx:tx + tp, :] for a in args[1:]]
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed