henry000 commited on
Commit
100c13d
Β·
1 Parent(s): add9e2f

πŸ› [Fix] bugs in onnx's output and video source

Browse files
yolo/tools/data_loader.py CHANGED
@@ -255,7 +255,7 @@ class StreamDataLoader:
255
  def load_video_file(self, video_path):
256
  import cv2
257
 
258
- cap = cv2.VideoCapture(video_path)
259
  while self.running:
260
  ret, frame = cap.read()
261
  if not ret:
 
255
  def load_video_file(self, video_path):
256
  import cv2
257
 
258
+ cap = cv2.VideoCapture(str(video_path))
259
  while self.running:
260
  ret, frame = cap.read()
261
  if not ret:
yolo/utils/deploy_utils.py CHANGED
@@ -45,6 +45,8 @@ class FastModelLoader:
45
  if idx % 3 == 2:
46
  model_outputs.append(layer_output)
47
  layer_output = []
 
 
48
  return {"Main": model_outputs}
49
 
50
  InferenceSession.__call__ = onnx_forward
 
45
  if idx % 3 == 2:
46
  model_outputs.append(layer_output)
47
  layer_output = []
48
+ if len(model_outputs) == 6:
49
+ model_outputs = model_outputs[:3]
50
  return {"Main": model_outputs}
51
 
52
  InferenceSession.__call__ = onnx_forward