From 3bf121db8c678a5b8ae955fd9ecc6cb5e94317b0 Mon Sep 17 00:00:00 2001 From: sosokker Date: Mon, 13 May 2024 15:24:08 +0700 Subject: [PATCH] Update action_model.py --- StreamServer/src/analytic/action/action_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/StreamServer/src/analytic/action/action_model.py b/StreamServer/src/analytic/action/action_model.py index c4928fc..7811429 100644 --- a/StreamServer/src/analytic/action/action_model.py +++ b/StreamServer/src/analytic/action/action_model.py @@ -105,9 +105,8 @@ class ActionModel: pts = np.array(track.keypoints_list, dtype=np.float32) out = self.action_model.predict(pts, frame.shape[:2]) action_name = self.action_model.class_names[out[0].argmax()] - action = '{}: {:.2f}%'.format(action_name, out[0].max() * 100) # Add action to action list. - ActionModel.ACTION_LIST.append(action) + ActionModel.ACTION_LIST.append(action_name) if action_name == 'Fall Down': ActionModel.IS_FALL_DOWN = True