Convolutional Neural Network for Classifying Yoga Poses

Dataset

Folder structure

cnn_multiclass.py 
cnn_multiclass_predict.py
/training_set/
/bridge/
TrainBrFile1.jpg
TrainBrFile2.jpg
...
/mountain/
TrainMoFile1.jpg
TrainMoFile2.jpg
...
...
/test_set/
/bridge/
TestBrFile1.jpg
TestBrFile2.jpg
...
/mountain/
TestMoFile1.jpg
TestMoFile2.jpg
...
...

What image types can I put in?

How many images should you have in the training set vs test set?

Data augmentation

When should data augmentation happen?

Model and training

classifier.add(Dense(units = 10, activation = 'softmax'))

Saving the model

classifier.save('my_new_model.h5')
from keras.models import load_model import h5py

Predictor

classifier = load_model('my_new_model.h5')
result = classifier.predict(test_image)

Quantum computing is everything. Quantum Physicist. @GeorgiaTech ‘12. Researcher @Bleximo. Let’s chat #quantum, #AI, and #womenintech. www.amarchenkova.com

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store