basilshaji commited on
Commit
8a7d30f
1 Parent(s): 650ad4b

Update yolov5/models/experimental.py

Browse files
Files changed (1) hide show
  1. yolov5/models/experimental.py +1 -1
yolov5/models/experimental.py CHANGED
@@ -91,7 +91,7 @@ def attempt_load(weights, device=None, inplace=True, fuse=True):
91
 
92
  Example inputs: weights=[a,b,c] or a single model weights=[a] or weights=a.
93
  """
94
- from models.yolo import Detect, Model
95
 
96
  model = Ensemble()
97
  for w in weights if isinstance(weights, list) else [weights]:
 
91
 
92
  Example inputs: weights=[a,b,c] or a single model weights=[a] or weights=a.
93
  """
94
+ from yolov5.models.yolo import Detect, Model
95
 
96
  model = Ensemble()
97
  for w in weights if isinstance(weights, list) else [weights]: