🕺Add models

ArthurFDLR · Jul 23, 2021 · 87cae92 · 87cae92
1 parent 1c1a7fb
commit 87cae92
Show file tree

Hide file tree

Showing 11 changed files with 4 additions and 2 deletions.
diff --git a/pose_classification_kit/models/Body/20Class_CNN_Augmented_BODY25/CNN-Augmented_body25.h5 b/pose_classification_kit/models/Body/20Class_CNN_Augmented_BODY25/CNN-Augmented_body25.h5
diff --git a/pose_classification_kit/models/Body/20Class_CNN_Augmented_BODY25/class.txt b/pose_classification_kit/models/Body/20Class_CNN_Augmented_BODY25/class.txt
@@ -0,0 +1 @@
+Seated,Stand,Stand_RightArmRaised,Stand_LeftArmRaised,T,MilitarySalute,PushUp_Low,Squat,Plank,Yoga_Tree_left,Yoga_Tree_right,Yoga_UpwardSalute,Yoga_Warrior2_left,Yoga_Warrior2_right,Traffic_AllStop,Traffic_BackStop,Traffic_FrontStop,Traffic_BackFrontStop,Traffic_LeftTurn,Traffic_RightTurn
diff --git a/...fication_kit/models/Body/20Class_CNN_BODY25/CNN-2Conv1D-64x3filter-2dense-2x128_body25.h5 b/...fication_kit/models/Body/20Class_CNN_BODY25/CNN-2Conv1D-64x3filter-2dense-2x128_body25.h5
diff --git a/pose_classification_kit/models/Body/20Class_CNN_BODY25/class.txt b/pose_classification_kit/models/Body/20Class_CNN_BODY25/class.txt
@@ -0,0 +1 @@
+Seated,Stand,Stand_RightArmRaised,Stand_LeftArmRaised,T,MilitarySalute,PushUp_Low,Squat,Plank,Yoga_Tree_left,Yoga_Tree_right,Yoga_UpwardSalute,Yoga_Warrior2_left,Yoga_Warrior2_right,Traffic_AllStop,Traffic_BackStop,Traffic_FrontStop,Traffic_BackFrontStop,Traffic_LeftTurn,Traffic_RightTurn
diff --git a/...asses_3x64_BODY18/9Classes_3x64_body18.h5 → ...Class_3x64_BODY18/9Classes_3x64_body18.h5 b/...asses_3x64_BODY18/9Classes_3x64_body18.h5 → ...Class_3x64_BODY18/9Classes_3x64_body18.h5
diff --git a/...odels/Body/9Classes_3x64_BODY18/class.txt → .../models/Body/9Class_3x64_BODY18/class.txt b/...odels/Body/9Classes_3x64_BODY18/class.txt → .../models/Body/9Class_3x64_BODY18/class.txt
diff --git a/...asses_3x64_BODY25/9Classes_3x64_body25.h5 → ...Class_3x64_BODY25/9Classes_3x64_body25.h5 b/...asses_3x64_BODY25/9Classes_3x64_body25.h5 → ...Class_3x64_BODY25/9Classes_3x64_body25.h5
diff --git a/...odels/Body/9Classes_3x64_BODY25/class.txt → .../models/Body/9Class_3x64_BODY25/class.txt b/...odels/Body/9Classes_3x64_BODY25/class.txt → .../models/Body/9Class_3x64_BODY25/class.txt
diff --git a/...2Conv1D-64x3filter-2dense-2x128_body25.h5 → ...2Conv1D-64x3filter-2dense-2x128_body25.h5 b/...2Conv1D-64x3filter-2dense-2x128_body25.h5 → ...2Conv1D-64x3filter-2dense-2x128_body25.h5
diff --git a/...models/Body/9Classes_CNN_BODY25/class.txt → ...t/models/Body/9Class_CNN_BODY25/class.txt b/...models/Body/9Classes_CNN_BODY25/class.txt → ...t/models/Body/9Class_CNN_BODY25/class.txt
diff --git a/pose_classification_kit/src/keypoints_analysis/body_analysis.py b/pose_classification_kit/src/keypoints_analysis/body_analysis.py
@@ -151,8 +151,8 @@ def updatePredictedClass(self, keypoints: np.ndarray):
 
             if self.modelClassifier is not None:
                 try:
-                    prediction = self.modelClassifier.predict(np.array([inputData]))[0]
-                    # prediction = self.modelClassifier.predict(np.array([keypoints.T[:,:2]]))[0]
+                    #prediction = self.modelClassifier.predict(np.array([inputData]))[0]
+                    prediction = self.modelClassifier.predict(np.array([keypoints.T[:,:2]]))[0]
                 except ValueError:
                     print("Model input shape not supported")
                 self.currentPrediction = self.classOutputs[np.argmax(prediction)]
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Seated,Stand,Stand_RightArmRaised,Stand_LeftArmRaised,T,MilitarySalute,PushUp_Low,Squat,Plank,Yoga_Tree_left,Yoga_Tree_right,Yoga_UpwardSalute,Yoga_Warrior2_left,Yoga_Warrior2_right,Traffic_AllStop,Traffic_BackStop,Traffic_FrontStop,Traffic_BackFrontStop,Traffic_LeftTurn,Traffic_RightTurn