ok
Browse files- 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(
|
|
|
|
| 61 |
"smpl",
|
| 62 |
-
str(
|
| 63 |
]
|
| 64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|