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

Refactor Python code (#13448)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-06-09 02:32:17 +02:00 • committed by GitHub
parent 6a234f3639
commit 1b26838def
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
22 changed files with 81 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/solutions/ai_gym.py
View file

@ -93,7 +93,7 @@ class AIGym:
self.stage[ind] = "up"
self.count[ind] += 1
elif self.pose_type == "pushup" or self.pose_type == "squat":
elif self.pose_type in {"pushup", "squat"}:
if self.angle[ind] > self.poseup_angle:
self.stage[ind] = "up"
if self.angle[ind] < self.posedown_angle and self.stage[ind] == "up":

2
ultralytics/solutions/object_counter.py
View file

@ -172,7 +172,7 @@ class ObjectCounter:
if self.draw_tracks:
self.annotator.draw_centroid_and_tracks(
track_line,
color=self.track_color if self.track_color else colors(int(track_id), True),
color=self.track_color or colors(int(track_id), True),
track_thickness=self.track_thickness,
)