search

zhjy2016 / SPLUT

Official code for ECCV2022 paper: Learning Series-Parallel Lookup Tables for Efficient Image Super-Resolution
50 stars 10 forks source link

When we try to train the SPLUT, but got this problem #3

Open StevenTony611 opened 2 years ago

StevenTony611 commented 2 years ago

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!

zhjy2016 commented 2 years ago

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!