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

ultralytics 8.2.5 New 🌟 Parking Management Solution (#10385)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-04-29 13:58:12 +05:00 • committed by GitHub
parent 156b6be8d3
commit bc9fd45cdf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 451 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

35
ultralytics/solutions/object_counter.py
View file

@ -181,9 +181,7 @@ class ObjectCounter:
# Store class info
if self.names[cls] not in self.class_wise_count:
if len(self.names[cls]) > 5:
self.names[cls] = self.names[cls][:5]
self.class_wise_count[self.names[cls]] = {"in": 0, "out": 0}
self.class_wise_count[self.names[cls]] = {"IN": 0, "OUT": 0}
# Draw Tracks
track_line = self.track_history[track_id]
@ -210,10 +208,10 @@ class ObjectCounter:
if (box[0] - prev_position[0]) * (self.counting_region.centroid.x - prev_position[0]) > 0:
self.in_counts += 1
self.class_wise_count[self.names[cls]]["in"] += 1
self.class_wise_count[self.names[cls]]["IN"] += 1
else:
self.out_counts += 1
self.class_wise_count[self.names[cls]]["out"] += 1
self.class_wise_count[self.names[cls]]["OUT"] += 1
# Count objects using line
elif len(self.reg_pts) == 2:
@ -224,33 +222,26 @@ class ObjectCounter:
if (box[0] - prev_position[0]) * (self.counting_region.centroid.x - prev_position[0]) > 0:
self.in_counts += 1
self.class_wise_count[self.names[cls]]["in"] += 1
self.class_wise_count[self.names[cls]]["IN"] += 1
else:
self.out_counts += 1
self.class_wise_count[self.names[cls]]["out"] += 1
self.class_wise_count[self.names[cls]]["OUT"] += 1
label = "Ultralytics Analytics \t"
labels_dict = {}
for key, value in self.class_wise_count.items():
if value["in"] != 0 or value["out"] != 0:
if value["IN"] != 0 or value["OUT"] != 0:
if not self.view_in_counts and not self.view_out_counts:
label = None
continue
elif not self.view_in_counts:
label += f"{str.capitalize(key)}: IN {value['in']} \t"
labels_dict[str.capitalize(key)] = f"OUT {value['OUT']}"
elif not self.view_out_counts:
label += f"{str.capitalize(key)}: OUT {value['out']} \t"
labels_dict[str.capitalize(key)] = f"IN {value['IN']}"
else:
label += f"{str.capitalize(key)}: IN {value['in']} OUT {value['out']} \t"
labels_dict[str.capitalize(key)] = f"IN {value['IN']} OUT {value['OUT']}"
label = label.rstrip()
label = label.split("\t")
if label is not None:
self.annotator.display_counts(
counts=label,
count_txt_color=self.count_txt_color,
count_bg_color=self.count_bg_color,
)
if labels_dict is not None:
self.annotator.display_analytics(self.im0, labels_dict, self.count_txt_color, self.count_bg_color, 10)
def display_frames(self):
"""Display frame."""