blitzkrieg0000 commited on
Commit
e16aa3e
·
verified ·
1 Parent(s): 4db6c0b

Update Lib/Core.py

Browse files
Files changed (1) hide show
  1. 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 = "./weight/yolov9c-cable-seg.pt"
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)