(data_root, split)
| 2 | import os |
| 3 | |
| 4 | def get_image_list(data_root, split): |
| 5 | filelist = [] |
| 6 | |
| 7 | with open('filelists/{}.txt'.format(split)) as f: |
| 8 | for line in f: |
| 9 | line = line.strip() |
| 10 | if ' ' in line: line = line.split()[0] |
| 11 | filelist.append(os.path.join(data_root, line)) |
| 12 | |
| 13 | return filelist |
| 14 | |
| 15 | class HParams: |
| 16 | def __init__(self, **kwargs): |