*(DASR) # ./main.sh
./data_set/benchmark/Set14/HR
./data_set/benchmark/Set14/LR_bicubic
Making model...
Preparing loss function:
1.000 L1
[Epoch 1] Learning rate: 1.00e-3
Traceback (most recent call last):
File "main.py", line 19, in
t.train()
File "/home/jovyan/sr/DASR-main/trainer.py", line 63, in train
for batch, (hr, _, idx_scale) in enumerate(self.loader_train):
File "/home/jovyan/sr/DASR/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "/home/jovyan/sr/DASR/lib/python3.6/site-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 "/home/jovyan/sr/DASR-main/dataloader.py", line 38, in _ms_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/home/jovyan/sr/DASR-main/dataloader.py", line 38, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/home/jovyan/sr/DASR-main/data/multiscalesrdata.py", line 123, in getitem
hr = self.get_patch(hr)
File "/home/jovyan/sr/DASR-main/data/multiscalesrdata.py", line 164, in get_patch
hr = common.augment(hr) if not self.args.no_augment else hr
File "/home/jovyan/sr/DASR-main/data/common.py", line 44, in augment
if hflip: img = img[:, ::-1, :]
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed**
非常感谢您的分享,但是我在测试自己的训练集时遇到一个问题,我的数据集是灰度图片,猜测应该是矩阵维数错误,可是我没有找到读取图片的代码在哪里,应该如何修改代码?感谢!
*(DASR) # ./main.sh ./data_set/benchmark/Set14/HR ./data_set/benchmark/Set14/LR_bicubic Making model... Preparing loss function: 1.000 L1 [Epoch 1] Learning rate: 1.00e-3 Traceback (most recent call last): File "main.py", line 19, in
t.train()
File "/home/jovyan/sr/DASR-main/trainer.py", line 63, in train
for batch, (hr, _, idx_scale) in enumerate(self.loader_train):
File "/home/jovyan/sr/DASR/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "/home/jovyan/sr/DASR/lib/python3.6/site-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 "/home/jovyan/sr/DASR-main/dataloader.py", line 38, in _ms_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/home/jovyan/sr/DASR-main/dataloader.py", line 38, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/home/jovyan/sr/DASR-main/data/multiscalesrdata.py", line 123, in getitem
hr = self.get_patch(hr)
File "/home/jovyan/sr/DASR-main/data/multiscalesrdata.py", line 164, in get_patch
hr = common.augment(hr) if not self.args.no_augment else hr
File "/home/jovyan/sr/DASR-main/data/common.py", line 44, in augment
if hflip: img = img[:, ::-1, :]
IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed**