Skip to content

Commit 5871965

Browse files
Update yolo_detect_and_count.py
1 parent 3b5b7da commit 5871965

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

yolo_detect_and_count.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -439,7 +439,7 @@ def predict_video(self, video_path, save_dir, save_format = "avi",
439439
frame = self.default_display(**display_args)
440440

441441
elif display == 'custom':
442-
frame == self.custom_display(colors, **display_args)
442+
frame == self.custom_display( **display_args)
443443

444444
# Display FPS on frame
445445
frame = cv2.putText(frame,f"FPS : {fps:,.2f}" ,

0 commit comments

Comments
 (0)