Spaces:
Sleeping
Sleeping
Update octolyzer/segment/sloseg/slo_inference.py
Browse files
octolyzer/segment/sloseg/slo_inference.py
CHANGED
|
@@ -107,7 +107,7 @@ def get_img_list_dataloader(img_list, batch_size=16, num_workers=0, pin_memory=F
|
|
| 107 |
|
| 108 |
class SLOSegmenter:
|
| 109 |
|
| 110 |
-
DEFAULT_MODEL_URL = 'https://huggingface.co/spaces/leosheck/SLO_vessel_analysis/resolve/main/weights/slosegmenter_weights.pth'
|
| 111 |
DEFAULT_THRESHOLD = 0.5
|
| 112 |
DEFAULT_MODEL_PATH = None
|
| 113 |
# DEFAULT_MODEL_PATH = os.path.join(SCRIPT_PATH, r"weights/slosegmenter_weights.pth")
|
|
|
|
| 107 |
|
| 108 |
class SLOSegmenter:
|
| 109 |
|
| 110 |
+
DEFAULT_MODEL_URL = 'https://huggingface.co/spaces/leosheck/SLO_vessel_analysis/resolve/main/octolyzer/segment/sloseg/weights/slosegmenter_weights.pth'
|
| 111 |
DEFAULT_THRESHOLD = 0.5
|
| 112 |
DEFAULT_MODEL_PATH = None
|
| 113 |
# DEFAULT_MODEL_PATH = os.path.join(SCRIPT_PATH, r"weights/slosegmenter_weights.pth")
|