sosokker/HomieCare
1
0
Fork 0
mirror of https://github.com/Sosokker/HomieCare.git synced 2025-12-19 02:04:03 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update action_model.py

Browse Source
This commit is contained in:
sosokker 2024-05-13 15:24:08 +07:00
parent eeb29dfd90
commit 3bf121db8c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
StreamServer/src/analytic/action/action_model.py
View File

@ -105,9 +105,8 @@ class ActionModel:
pts = np.array(track.keypoints_list, dtype=np.float32) pts = np.array(track.keypoints_list, dtype=np.float32)
out = self.action_model.predict(pts, frame.shape[:2]) out = self.action_model.predict(pts, frame.shape[:2])
action_name = self.action_model.class_names[out[0].argmax()] action_name = self.action_model.class_names[out[0].argmax()]
action = '{}: {:.2f}%'.format(action_name, out[0].max() * 100)
# Add action to action list. # Add action to action list.
ActionModel.ACTION_LIST.append(action) ActionModel.ACTION_LIST.append(action_name)
if action_name == 'Fall Down': if action_name == 'Fall Down':
ActionModel.IS_FALL_DOWN = True ActionModel.IS_FALL_DOWN = True