Fix notebook `classes_names` argument with `names` (#15915)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/15932/head
Muhammad Rizwan Munawar 3 months ago committed by GitHub
parent 885a56837f
commit eecee8f666
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      examples/heatmaps.ipynb
  2. 2
      examples/object_counting.ipynb

@ -116,7 +116,7 @@
" colormap=cv2.COLORMAP_PARULA,\n", " colormap=cv2.COLORMAP_PARULA,\n",
" view_img=True,\n", " view_img=True,\n",
" shape=\"circle\",\n", " shape=\"circle\",\n",
" classes_names=model.names,\n", " names=model.names,\n",
")\n", ")\n",
"\n", "\n",
"while cap.isOpened():\n", "while cap.isOpened():\n",

@ -129,7 +129,7 @@
"counter = solutions.ObjectCounter(\n", "counter = solutions.ObjectCounter(\n",
" view_img=True, # Display the image during processing\n", " view_img=True, # Display the image during processing\n",
" reg_pts=line_points, # Region of interest points\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", " draw_tracks=True, # Draw tracking lines for objects\n",
" line_thickness=2, # Thickness of the lines drawn\n", " line_thickness=2, # Thickness of the lines drawn\n",
")\n", ")\n",

Loading…
Cancel
Save