xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 06:14:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update the heatmap module code + docs (#7045)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2023-12-18 20:47:32 +05:00 committed by GitHub
parent 34b10b2db3
commit 1d0da53a32
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 24 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/en/guides/heatmaps.md
View File

@ -44,7 +44,8 @@ A heatmap generated with [Ultralytics YOLOv8](https://github.com/ultralytics/ult
heatmap_obj.set_args(colormap=cv2.COLORMAP_CIVIDIS, heatmap_obj.set_args(colormap=cv2.COLORMAP_CIVIDIS,
imw=cap.get(4), # should same as cap width imw=cap.get(4), # should same as cap width
imh=cap.get(3), # should same as cap height imh=cap.get(3), # should same as cap height
view_img=True) view_img=True,
decay_factor=0.99)
while cap.isOpened(): while cap.isOpened():
success, im0 = cap.read() success, im0 = cap.read()
@ -184,7 +185,7 @@ A heatmap generated with [Ultralytics YOLOv8](https://github.com/ultralytics/ult
### Arguments `set_args` ### Arguments `set_args`
| Name | Type | Default | Description | | Name | Type | Default | Description |
|---------------------|----------------|-----------------|---------------------------------| |---------------------|----------------|-----------------|-----------------------------------------------------------|
| view_img | `bool` | `False` | Display the frame with heatmap | | view_img | `bool` | `False` | Display the frame with heatmap |
| colormap | `cv2.COLORMAP` | `None` | cv2.COLORMAP for heatmap | | colormap | `cv2.COLORMAP` | `None` | cv2.COLORMAP for heatmap |
| imw | `int` | `None` | Width of Heatmap | | imw | `int` | `None` | Width of Heatmap |
@ -194,6 +195,7 @@ A heatmap generated with [Ultralytics YOLOv8](https://github.com/ultralytics/ult
| count_txt_thickness | `int` | `2` | Count values text size | | count_txt_thickness | `int` | `2` | Count values text size |
| count_reg_color | `tuple` | `(255, 0, 255)` | Counting region color | | count_reg_color | `tuple` | `(255, 0, 255)` | Counting region color |
| region_thickness | `int` | `5` | Counting region thickness value | | region_thickness | `int` | `5` | Counting region thickness value |
| decay_factor | `float` | `0.99` | Decay factor for heatmap area removal after specific time |
### Arguments `model.track` ### Arguments `model.track`

9
ultralytics/solutions/heatmap.py
View File

@ -50,6 +50,9 @@ class Heatmap:
self.count_reg_color = (0, 255, 0) self.count_reg_color = (0, 255, 0)
self.region_thickness = 5 self.region_thickness = 5
# Decay factor
self.decay_factor = 0.99
# Check if environment support imshow # Check if environment support imshow
self.env_check = check_imshow(warn=True) self.env_check = check_imshow(warn=True)
@ -62,7 +65,8 @@ class Heatmap:
count_reg_pts=None, count_reg_pts=None,
count_txt_thickness=2, count_txt_thickness=2,
count_reg_color=(255, 0, 255), count_reg_color=(255, 0, 255),
region_thickness=5): region_thickness=5,
decay_factor=0.99):
""" """
Configures the heatmap colormap, width, height and display parameters. Configures the heatmap colormap, width, height and display parameters.
@ -76,6 +80,7 @@ class Heatmap:
count_txt_thickness (int): Text thickness for object counting display count_txt_thickness (int): Text thickness for object counting display
count_reg_color (RGB color): Color of object counting region count_reg_color (RGB color): Color of object counting region
region_thickness (int): Object counting Region thickness region_thickness (int): Object counting Region thickness
decay_factor (float): value for removing heatmap area after object passed
""" """
self.imw = imw self.imw = imw
self.imh = imh self.imh = imh
@ -93,6 +98,7 @@ class Heatmap:
self.count_txt_thickness = count_txt_thickness # Counting text thickness self.count_txt_thickness = count_txt_thickness # Counting text thickness
self.count_reg_color = count_reg_color self.count_reg_color = count_reg_color
self.region_thickness = region_thickness self.region_thickness = region_thickness
self.decay_factor = decay_factor
def extract_results(self, tracks): def extract_results(self, tracks):
""" """
@ -117,6 +123,7 @@ class Heatmap:
if tracks[0].boxes.id is None: if tracks[0].boxes.id is None:
return self.im0 return self.im0
self.heatmap *= self.decay_factor # decay factor
self.extract_results(tracks) self.extract_results(tracks)
self.annotator = Annotator(self.im0, self.count_txt_thickness, None) self.annotator = Annotator(self.im0, self.count_txt_thickness, None)