Spaces:
Sleeping
Sleeping
Sean Carnahan
commited on
Commit
·
b39cc0b
1
Parent(s):
e2492f0
Incorporate external/BodybuildingPoseClassifier files directly, remove submodule
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- external/BodybuildingPoseClassifier/.gitignore +0 -0
- external/BodybuildingPoseClassifier/LICENSE +21 -0
- external/BodybuildingPoseClassifier/README.md +36 -0
- external/BodybuildingPoseClassifier/app.py +39 -0
- external/BodybuildingPoseClassifier/bodybuilding_pose_classifier.h5 +3 -0
- external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/1.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/2.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/3.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/4.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/5.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bLatSpread/1.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bLatSpread/2.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bLatSpread/3.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bLatSpread/4.png +3 -0
- external/BodybuildingPoseClassifier/data/train/bLatSpread/5.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/1.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/2.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/3.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/4.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/5.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fLatSpread/1.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fLatSpread/2.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fLatSpread/3.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fLatSpread/4.png +3 -0
- external/BodybuildingPoseClassifier/data/train/fLatSpread/5.png +3 -0
- external/BodybuildingPoseClassifier/data/train/sideChest/1.png +3 -0
- external/BodybuildingPoseClassifier/data/train/sideChest/2.png +3 -0
- external/BodybuildingPoseClassifier/data/train/sideChest/3.png +3 -0
- external/BodybuildingPoseClassifier/data/train/sideChest/4.png +3 -0
- external/BodybuildingPoseClassifier/data/train/sideChest/5.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/1.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/2.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/3.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/4.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/5.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bLatSpread/1.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bLatSpread/2.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bLatSpread/3.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bLatSpread/4.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/bLatSpread/5.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/1.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/2.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/3.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/4.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/5.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fLatSpread/1.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fLatSpread/2.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fLatSpread/3.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fLatSpread/4.png +3 -0
- external/BodybuildingPoseClassifier/data/validation/fLatSpread/5.png +3 -0
external/BodybuildingPoseClassifier/.gitignore
ADDED
Binary file (68 Bytes). View file
|
|
external/BodybuildingPoseClassifier/LICENSE
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MIT License
|
2 |
+
|
3 |
+
Copyright (c) 2024 David Unterguggenberger
|
4 |
+
|
5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
6 |
+
of this software and associated documentation files (the "Software"), to deal
|
7 |
+
in the Software without restriction, including without limitation the rights
|
8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
9 |
+
copies of the Software, and to permit persons to whom the Software is
|
10 |
+
furnished to do so, subject to the following conditions:
|
11 |
+
|
12 |
+
The above copyright notice and this permission notice shall be included in all
|
13 |
+
copies or substantial portions of the Software.
|
14 |
+
|
15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
21 |
+
SOFTWARE.
|
external/BodybuildingPoseClassifier/README.md
ADDED
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Bodybuilding Pose Classifier
|
2 |
+
|
3 |
+
## Project Overview
|
4 |
+
This model is trained to identify and classify various bodybuilding poses from images. The aim of this project is to demonstrate the application of deep learning techniques for automated image recognition in specific sports disciplines.
|
5 |
+
|
6 |
+
## Model Description
|
7 |
+
The model uses a Convolutional Neural Network (CNN) architecture to recognize specific poses from images of bodybuilders. It has been trained with training data and validated with validation data to ensure the accuracy and robustness of the model.
|
8 |
+
|
9 |
+
## Datasets
|
10 |
+
- **Training Data**: Consists of 25 images per pose category, captured from different angles and under various lighting conditions.
|
11 |
+
- **Validation Data**: Used to monitor the model's performance during training and to prevent overfitting. Contains 25 images per pose category, similar but not identical to the training images.
|
12 |
+
|
13 |
+
## Technical Details
|
14 |
+
The model was developed in Python using the following libraries and technologies:
|
15 |
+
|
16 |
+
- **TensorFlow and Keras**: For constructing, training, and validating the CNN.
|
17 |
+
- **NumPy**: For data manipulation and processing.
|
18 |
+
- **Matplotlib**: For visualizing training and validation results.
|
19 |
+
- **ImageDataGenerator from Keras**: Used for image preprocessing and data augmentation to enhance model performance by increasing the diversity of the training data.
|
20 |
+
|
21 |
+
## Installation
|
22 |
+
1. Clone the repository:
|
23 |
+
```bash
|
24 |
+
git clone https://github.com/JumpiiX/BodybuildingPoseClassifier.git
|
25 |
+
```
|
26 |
+
2. Install the required dependencies:
|
27 |
+
```bash
|
28 |
+
pip install -r requirements.txt
|
29 |
+
```
|
30 |
+
|
31 |
+
## Usage
|
32 |
+
|
33 |
+
### Train and Test the Model
|
34 |
+
To train and test the model, execute the following command:
|
35 |
+
```bash
|
36 |
+
python training/train_model.py
|
external/BodybuildingPoseClassifier/app.py
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from flask import Flask, request, render_template
|
2 |
+
from tensorflow.keras.models import load_model
|
3 |
+
from tensorflow.keras.preprocessing import image
|
4 |
+
import numpy as np
|
5 |
+
import os
|
6 |
+
|
7 |
+
app = Flask(__name__)
|
8 |
+
|
9 |
+
model_path = 'bodybuilding_pose_classifier.h5'
|
10 |
+
|
11 |
+
model = load_model(model_path)
|
12 |
+
|
13 |
+
class_labels = ['Side Chest', 'Front Double Biceps', 'Back Double Biceps', 'Front Lat Spread', 'Back Lat Spread']
|
14 |
+
|
15 |
+
def predict_pose(img_path):
|
16 |
+
img = image.load_img(img_path, target_size=(150, 150))
|
17 |
+
img_array = image.img_to_array(img)
|
18 |
+
img_array = np.expand_dims(img_array, axis=0) / 255.0
|
19 |
+
predictions = model.predict(img_array)
|
20 |
+
predicted_class = np.argmax(predictions, axis=1)
|
21 |
+
return class_labels[predicted_class[0]]
|
22 |
+
|
23 |
+
@app.route('/', methods=['GET', 'POST'])
|
24 |
+
def upload_file():
|
25 |
+
if request.method == 'POST':
|
26 |
+
if 'file' not in request.files:
|
27 |
+
return 'No file part'
|
28 |
+
file = request.files['file']
|
29 |
+
if file.filename == '':
|
30 |
+
return 'No selected file'
|
31 |
+
if file:
|
32 |
+
filepath = os.path.join('uploads', file.filename)
|
33 |
+
file.save(filepath)
|
34 |
+
pose = predict_pose(filepath)
|
35 |
+
return render_template('result.html', pose=pose)
|
36 |
+
return render_template('index.html')
|
37 |
+
|
38 |
+
if __name__ == '__main__':
|
39 |
+
app.run(debug=True)
|
external/BodybuildingPoseClassifier/bodybuilding_pose_classifier.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:56cdfbdadbef2675622e699fcf444d5bcb0aab6c695bb32165ae60e984278346
|
3 |
+
size 228483160
|
external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bDoubleBiceps/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bLatSpread/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bLatSpread/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bLatSpread/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bLatSpread/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/bLatSpread/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fDoubleBiceps/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fLatSpread/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fLatSpread/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fLatSpread/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fLatSpread/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/fLatSpread/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/sideChest/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/sideChest/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/sideChest/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/sideChest/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/train/sideChest/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bDoubleBiceps/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bLatSpread/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bLatSpread/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bLatSpread/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bLatSpread/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/bLatSpread/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fDoubleBiceps/5.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fLatSpread/1.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fLatSpread/2.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fLatSpread/3.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fLatSpread/4.png
ADDED
![]() |
Git LFS Details
|
external/BodybuildingPoseClassifier/data/validation/fLatSpread/5.png
ADDED
![]() |
Git LFS Details
|