blitzkrieg0000 commited on
Commit
c573583
·
verified ·
1 Parent(s): ad26028

Update Lib/Core.py

Browse files
Files changed (1) hide show
  1. 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/yolov8l-seg-pre100.onnx"
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