debug

owl_core.py

@@ -117,8 +117,9 @@ def owl_full_video(
             df = pd.concat([df, row], ignore_index=True)
             
             # if there are detections, save the frame replacing the original frame
-            annotated_frame = plot_predictions(image, labels[j], scores, boxes)
-            cv2.imwrite(image, annotated_frame)
+            if len(labels[j]) > 0:
+                annotated_frame = plot_predictions(image, labels[j], scores, boxes)
+                cv2.imwrite(image, annotated_frame)
                 
     # save the df to a csv
     csv_path = f"{results_dir}/{filename}_{threshold}.csv"