annayding commited on
Commit
6f14479
·
1 Parent(s): bfad7a6
Files changed (1) hide show
  1. owl_core.py +1 -1
owl_core.py CHANGED
@@ -117,7 +117,7 @@ def owl_full_video(
117
  df = pd.concat([df, row], ignore_index=True)
118
 
119
  # if there are detections, save the frame replacing the original frame
120
- if len(labels[j]) > 0:
121
  annotated_frame = plot_predictions(image, labels[j], scores, boxes)
122
  cv2.imwrite(image, annotated_frame)
123
 
 
117
  df = pd.concat([df, row], ignore_index=True)
118
 
119
  # if there are detections, save the frame replacing the original frame
120
+ if labels[j] is not None:
121
  annotated_frame = plot_predictions(image, labels[j], scores, boxes)
122
  cv2.imwrite(image, annotated_frame)
123