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",