liuh127 / NTIRE-2021-Dehazing-Two-branch

Official PyTorch implementation of Two-branch Dehazing, wining runner-up awards in NTIRE 2021.
MIT License
50 stars 7 forks source link

ABOUT TRAIN #9

Open EKChloe opened 2 years ago

EKChloe commented 2 years ago

in train_dataset.py
for line in open(os.path.join(train_dir, 'train.txt')): line = line.strip('\n') if line!='': self.list_train.append(line) if i want to re-train this work on SOTS or I-HAZE,do i need to build this train.txt file?