Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Don't bind resolver

This commit is contained in:
Joel Holdbrooks 2023-01-24 08:36:40 -08:00
parent a7505df0a6
commit 94e9abf184
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
data/resolver.py
View File

@ -191,16 +191,12 @@ def resolve_root(path: str, aspects: list[float], flip_p: float = 0.0, seed=555)
:param flip_p: The probability of flipping the image
"""
if os.path.isfile(path) and path.endswith('.json'):
resolver = JSONResolver(aspects, flip_p, seed)
return JSONResolver(aspects, flip_p, seed).image_train_items(path)
if os.path.isdir(path):
resolver = DirectoryResolver(aspects, flip_p, seed)
return DirectoryResolver(aspects, flip_p, seed).image_train_items(path)
if not resolver:
raise ValueError(f"data_root '{path}' is not a valid directory or JSON file.")
items = resolver.image_train_items(path)
return items
raise ValueError(f"data_root '{path}' is not a valid directory or JSON file.")
def resolve(value: typing.Union[dict, str], aspects: list[float], flip_p: float=0.0, seed=555) -> list[ImageTrainItem]:
"""