HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix notebook classes_names argument with names (#15915)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-08-31 20:35:59 +05:00 • committed by GitHub
parent 885a56837f
commit eecee8f666
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/heatmaps.ipynb
View file

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

2
examples/object_counting.ipynb
View file

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