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

ultralytics 8.2.53 Heatmaps fix for empty images (#14329)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: RizwanMunawar <chr043416@gmail.com>
This commit is contained in:
Francesco Mattioli 2024-07-10 23:38:20 +02:00 • committed by GitHub
parent 2d332a1cb1
commit 95736975fc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/__init__.py
View file

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.2.52"
__version__ = "8.2.53"
import os

17
ultralytics/solutions/heatmap.py
View file

@ -60,9 +60,9 @@ class Heatmap:
self.heatmap_alpha = heatmap_alpha
# Predict/track information
self.boxes = None
self.track_ids = None
self.clss = None
self.boxes = []
self.track_ids = []
self.clss = []
self.track_history = defaultdict(list)
# Region & Line Information
@ -107,16 +107,17 @@ class Heatmap:
print("Using Circular shape now")
self.shape = "circle"
def extract_results(self, tracks, _intialized=False):
def extract_results(self, tracks):
"""
Extracts results from the provided data.
Args:
tracks (list): List of tracks obtained from the object tracking process.
"""
self.boxes = tracks[0].boxes.xyxy.cpu()
self.clss = tracks[0].boxes.cls.cpu().tolist()
self.track_ids = tracks[0].boxes.id.int().cpu().tolist()
if tracks[0].boxes.id is not None:
self.boxes = tracks[0].boxes.xyxy.cpu()
self.clss = tracks[0].boxes.cls.tolist()
self.track_ids = tracks[0].boxes.id.int().tolist()
def generate_heatmap(self, im0, tracks):
"""
@ -138,7 +139,7 @@ class Heatmap:
self.extract_results(tracks)
self.annotator = Annotator(self.im0, self.tf, None)
if self.track_ids is not None:
if self.track_ids:
# Draw counting region
if self.count_reg_pts is not None:
self.annotator.draw_region(