Spaces:
Sleeping
Sleeping
Andres Felipe Ruiz-Hurtado
commited on
Commit
·
b710899
1
Parent(s):
32bf71b
fixes
Browse files- processsors.py +2 -0
processsors.py
CHANGED
|
@@ -94,6 +94,8 @@ class RootSegmentor():
|
|
| 94 |
|
| 95 |
width = 8
|
| 96 |
depth = 5
|
|
|
|
|
|
|
| 97 |
|
| 98 |
if self.model_type == "segroot":
|
| 99 |
#weights_path = os.path.join(r"D:\local_mydev\roots_finetuning\SegRoot0\weights\best_segnet-(8,5)-0.6441.pt"
|
|
|
|
| 94 |
|
| 95 |
width = 8
|
| 96 |
depth = 5
|
| 97 |
+
|
| 98 |
+
weights_path = os.path.join(MODELS_PATH, r"best_segnet-(8,5)-0.6441.pt")
|
| 99 |
|
| 100 |
if self.model_type == "segroot":
|
| 101 |
#weights_path = os.path.join(r"D:\local_mydev\roots_finetuning\SegRoot0\weights\best_segnet-(8,5)-0.6441.pt"
|