liuh127 / NTIRE-2021-Dehazing-Two-branch

Official PyTorch implementation of Two-branch Dehazing, wining runner-up awards in NTIRE 2021.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ABOUT TRAIN

EKChloe opened this issue · comments

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?