nexusbert commited on
Commit
cac11c8
·
1 Parent(s): 05ea946
Files changed (1) hide show
  1. smpl_generator.py +11 -3
smpl_generator.py CHANGED
@@ -56,12 +56,20 @@ class SMPLGenerator:
56
  if not models_dir.exists():
57
  models_dir = model_path_obj / "models"
58
 
 
59
  model_paths_to_try = [
60
- str(model_path_obj),
 
61
  "smpl",
62
- str(models_dir.parent) if models_dir.exists() else None
63
  ]
64
- model_paths_to_try = [p for p in model_paths_to_try if p is not None and Path(p).exists()]
 
 
 
 
 
 
65
 
66
  last_error = None
67
  for try_path in model_paths_to_try:
 
56
  if not models_dir.exists():
57
  models_dir = model_path_obj / "models"
58
 
59
+ base_path = Path(".").absolute()
60
  model_paths_to_try = [
61
+ str(base_path),
62
+ ".",
63
  "smpl",
64
+ str(model_path_obj),
65
  ]
66
+
67
+ if models_dir.exists():
68
+ parent_of_smpl = models_dir.parent.parent
69
+ if parent_of_smpl.exists():
70
+ model_paths_to_try.append(str(parent_of_smpl))
71
+
72
+ model_paths_to_try = list(dict.fromkeys(model_paths_to_try))
73
 
74
  last_error = None
75
  for try_path in model_paths_to_try: