MarkusEssl commited on
Commit
1d3d6a8
·
1 Parent(s): ad5c303
signify/yolo/models/experimental.py CHANGED
@@ -112,7 +112,7 @@ class Ensemble(nn.ModuleList):
112
 
113
 
114
  def attempt_load(weights, map_location=None, inplace=True):
115
- from models.yolo import Detect, Model
116
 
117
  # Loads an ensemble of models weights=[a,b,c] or a single model weights=[a] or weights=a
118
  model = Ensemble()
 
112
 
113
 
114
  def attempt_load(weights, map_location=None, inplace=True):
115
+ from signify.yolo.models.yolo import Detect, Model
116
 
117
  # Loads an ensemble of models weights=[a,b,c] or a single model weights=[a] or weights=a
118
  model = Ensemble()