When we try to train the SPLUT, but got this problem
StevenTony611 opened this issue · comments
File "/share4/home/zqiang/SPLUT-main/training_testing_code/utils.py", line 347, in next
self.lr_pngs, self.sharp_pngs, self.sr_pngs = self.shuffle_list(self.lr_pngs, self.sharp_pngs, self.sr_pngs)
ValueError: not enough values to unpack (expected 3, got 0)
===> Training start
0%| | 0/2000000 [00:00<?, ?it/s]
Traceback (most recent call last):
File "Train_SPLUT_S.py", line 230, in
batch_L, batch_H, batch_Stage1 = Iter_H.dequeue()
TypeError: cannot unpack non-iterable NoneType object
We don't know that the '/DIV2K_train_SR/*.png' includes the what type images in ./utils.py, so that cause this problem.
Thank you!
I checked the relevant codes. This part 'DIV2K_train_SR' is not referenced in the subsequent code. You can comment out this line or download the latest version of the code. Thanks for attention!