mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

f

Browse Source
This commit is contained in:
James Betker 2021-11-07 18:52:05 -07:00
parent 18b1de9b2c
commit 722d3dbdc2
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/util.py
View File

@ -590,7 +590,9 @@ def load_paths_from_cache(paths, cache_path, exclusion_list=[]):
output.extend(find_files_of_type('img', p, qualifier=is_audio_file)[0]) output.extend(find_files_of_type('img', p, qualifier=is_audio_file)[0])
if exclusion_list is not None and len(exclusion_list) > 0: if exclusion_list is not None and len(exclusion_list) > 0:
print(f"Removing exclusion lists..") print(f"Removing exclusion lists..")
before = len(output)
output = filter(lambda p: p not in exclusion_list, output) output = filter(lambda p: p not in exclusion_list, output)
print(f"Excluded {before-len(output)} files.")
print("Done.") print("Done.")
torch.save(output, cache_path) torch.save(output, cache_path)
return output return output