From eecee8f6660536e80fed613ec91ac6914f92de2a Mon Sep 17 00:00:00 2001 From: Muhammad Rizwan Munawar Date: Sat, 31 Aug 2024 20:35:59 +0500 Subject: [PATCH] Fix notebook `classes_names` argument with `names` (#15915) Co-authored-by: Glenn Jocher --- examples/heatmaps.ipynb | 2 +- examples/object_counting.ipynb | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/heatmaps.ipynb b/examples/heatmaps.ipynb index 6102e41cbd..1f590b8cb0 100644 --- a/examples/heatmaps.ipynb +++ b/examples/heatmaps.ipynb @@ -116,7 +116,7 @@ " colormap=cv2.COLORMAP_PARULA,\n", " view_img=True,\n", " shape=\"circle\",\n", - " classes_names=model.names,\n", + " names=model.names,\n", ")\n", "\n", "while cap.isOpened():\n", diff --git a/examples/object_counting.ipynb b/examples/object_counting.ipynb index 58e3d6cf48..8c3d0ba6e8 100644 --- a/examples/object_counting.ipynb +++ b/examples/object_counting.ipynb @@ -129,7 +129,7 @@ "counter = solutions.ObjectCounter(\n", " view_img=True, # Display the image during processing\n", " reg_pts=line_points, # Region of interest points\n", - " classes_names=model.names, # Class names from the YOLO model\n", + " names=model.names, # Class names from the YOLO model\n", " draw_tracks=True, # Draw tracking lines for objects\n", " line_thickness=2, # Thickness of the lines drawn\n", ")\n",