Update Lib/Core.py
Browse files- Lib/Core.py +1 -1
Lib/Core.py
CHANGED
|
@@ -8,7 +8,7 @@ from ultralytics.engine.results import Masks
|
|
| 8 |
class CablePoleSegmentation():
|
| 9 |
def __init__(self, MODEL_PATH=None, retina_mask=False):
|
| 10 |
if not MODEL_PATH:
|
| 11 |
-
MODEL_PATH = "./weight/
|
| 12 |
self._RetinaMask=retina_mask
|
| 13 |
self.Model = YOLO(MODEL_PATH) # load a custom model
|
| 14 |
|
|
|
|
| 8 |
class CablePoleSegmentation():
|
| 9 |
def __init__(self, MODEL_PATH=None, retina_mask=False):
|
| 10 |
if not MODEL_PATH:
|
| 11 |
+
MODEL_PATH = "./weight/yolov9c-cable-seg.pt"
|
| 12 |
self._RetinaMask=retina_mask
|
| 13 |
self.Model = YOLO(MODEL_PATH) # load a custom model
|
| 14 |
|