Update notebooks (#17260)

Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
pull/17252/head^2
Muhammad Rizwan Munawar 2 weeks ago committed by GitHub
parent 83404afff1
commit 6ffd8841fd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      examples/heatmaps.ipynb
  2. 2
      examples/object_counting.ipynb
  3. 2
      examples/object_tracking.ipynb

@ -112,7 +112,7 @@
"heatmap_obj = solutions.Heatmap(\n",
" colormap=cv2.COLORMAP_PARULA, # Color of the heatmap\n",
" show=True, # Display the image during processing\n",
" model=yolo11n.pt, # Ultralytics YOLO11 model file\n",
" model=\"yolo11n.pt\", # Ultralytics YOLO11 model file\n",
")\n",
"\n",
"while cap.isOpened():\n",

@ -123,7 +123,7 @@
"counter = solutions.ObjectCounter(\n",
" show=True, # Display the image during processing\n",
" region=line_points, # Region of interest points\n",
" model=yolo11n.pt, # Ultralytics YOLO11 model file\n",
" model=\"yolo11n.pt\", # Ultralytics YOLO11 model file\n",
" line_width=2, # Thickness of the lines and bounding boxes\n",
")\n",
"\n",

@ -176,7 +176,7 @@
"\n",
" # Annotate each mask with its corresponding tracking ID and color\n",
" for mask, track_id in zip(masks, track_ids):\n",
" annotator.seg_bbox(mask=mask, mask_color=colors(track_id, True), track_label=str(track_id))\n",
" annotator.seg_bbox(mask=mask, mask_color=colors(int(track_id), True), label=str(track_id))\n",
"\n",
" # Write the annotated frame to the output video\n",
" out.write(im0)\n",

Loading…
Cancel
Save