annayding
commited on
Commit
·
6f14479
1
Parent(s):
bfad7a6
debug
Browse files- 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
|
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 |
|