Update Lib/Core.py
Browse files- Lib/Core.py +1 -1
Lib/Core.py
CHANGED
|
@@ -16,7 +16,7 @@ DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
| 16 |
class CablePoleSegmentation():
|
| 17 |
def __init__(self, model_path=None, retina_mask=False):
|
| 18 |
if not model_path:
|
| 19 |
-
model_path = "./
|
| 20 |
self._RetinaMask=retina_mask
|
| 21 |
self.Model = None
|
| 22 |
self.PrepareModel(model_path)
|
|
|
|
| 16 |
class CablePoleSegmentation():
|
| 17 |
def __init__(self, model_path=None, retina_mask=False):
|
| 18 |
if not model_path:
|
| 19 |
+
model_path = "./Weight/yolov9c-cable-seg.pt"
|
| 20 |
self._RetinaMask=retina_mask
|
| 21 |
self.Model = None
|
| 22 |
self.PrepareModel(model_path)
|