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

ultralytics 8.0.239 Ultralytics Actions and hub-sdk adoption (#7431)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2024-01-10 03:16:08 +01:00 • committed by GitHub
parent e795277391
commit fe27db2f6e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
139 changed files with 6870 additions and 5125 deletions
Download patch file Download diff file Expand all files Collapse all files

29
ultralytics/solutions/speed_estimation.py
View file

@ -66,7 +66,7 @@ class SpeedEstimator:
spdl_dist_thresh (int): Euclidean distance threshold for speed line
"""
if reg_pts is None:
print('Region points not provided, using default values')
print("Region points not provided, using default values")
else:
self.reg_pts = reg_pts
self.names = names
@ -114,8 +114,9 @@ class SpeedEstimator:
cls (str): object class name
track (list): tracking history for tracks path drawing
"""
speed_label = str(int(
self.dist_data[track_id])) + 'km/ph' if track_id in self.dist_data else self.names[int(cls)]
speed_label = (
str(int(self.dist_data[track_id])) + "km/ph" if track_id in self.dist_data else self.names[int(cls)]
)
bbox_color = colors(int(track_id)) if track_id in self.dist_data else (255, 0, 255)
self.annotator.box_label(box, speed_label, bbox_color)
@ -132,19 +133,16 @@ class SpeedEstimator:
"""
if self.reg_pts[0][0] < track[-1][0] < self.reg_pts[1][0]:
if self.reg_pts[1][1] - self.spdl_dist_thresh < track[-1][1] < self.reg_pts[1][1] + self.spdl_dist_thresh:
direction = "known"
if (self.reg_pts[1][1] - self.spdl_dist_thresh < track[-1][1] < self.reg_pts[1][1] + self.spdl_dist_thresh):
direction = 'known'
elif (self.reg_pts[0][1] - self.spdl_dist_thresh < track[-1][1] <
self.reg_pts[0][1] + self.spdl_dist_thresh):
direction = 'known'
elif self.reg_pts[0][1] - self.spdl_dist_thresh < track[-1][1] < self.reg_pts[0][1] + self.spdl_dist_thresh:
direction = "known"
else:
direction = 'unknown'
if self.trk_previous_times[trk_id] != 0 and direction != 'unknown':
direction = "unknown"
if self.trk_previous_times[trk_id] != 0 and direction != "unknown":
if trk_id not in self.trk_idslist:
self.trk_idslist.append(trk_id)
@ -178,7 +176,6 @@ class SpeedEstimator:
self.annotator.draw_region(reg_pts=self.reg_pts, color=(255, 0, 0), thickness=self.region_thickness)
for box, trk_id, cls in zip(self.boxes, self.trk_ids, self.clss):
track = self.store_track_info(trk_id, box)
if trk_id not in self.trk_previous_times:
@ -194,10 +191,10 @@ class SpeedEstimator:
def display_frames(self):
"""Display frame."""
cv2.imshow('Ultralytics Speed Estimation', self.im0)
if cv2.waitKey(1) & 0xFF == ord('q'):
cv2.imshow("Ultralytics Speed Estimation", self.im0)
if cv2.waitKey(1) & 0xFF == ord("q"):
return
if __name__ == '__main__':
if __name__ == "__main__":
SpeedEstimator()