- Notifications
You must be signed in to change notification settings - Fork1
🎸 Machine Learning for Guitars: Retraining InceptionV3 to classify guitars and Converting Tensorflow to CoreML for iOS (presentation code)
License
jottenlips/Guitar-Classifier
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Machine Learning for Guitars - this is an elaboration on tensorflow for poets
pip install virtualenv
virtualenv env
source env/bin/activatepip install -tensorflowpip install -pillowpip install -tfcoremlIMAGE_SIZE=224ARCHITECTURE="mobilenet_0.50_${IMAGE_SIZE}"
I usedhttps://github.com/hardikvasa/google-images-download CLI
change the paths in resize.py
cd scriptspython resize.py
python -m scripts.retrain \ --bottleneck_dir=tf_files/bottlenecks \ --how_many_training_steps=500 \ --model_dir=tf_files/models/ \ --summaries_dir=tf_files/training_summaries/"${ARCHITECTURE}" \ --output_graph=tf_files/retrained_graph.pb \ --output_labels=tf_files/retrained_labels.txt \ --architecture="${ARCHITECTURE}" \ --image_dir=tf_files/Guitarspython -m scripts.label_image \ --graph=tf_files/retrained_graph.pb \ --image=../Desktop/sg.jpgtensorboard --logdir tf_files/training_summaries &
cd scripts
Change all paths to where you want to read/write your model and test your images.
python labelcoreml.py
python converter.py
From Core ML in ArKit, my fork from the presentation will be up soon.https://github.com/hanleyweng/CoreML-in-ARKit
https://github.com/jottenlips/GuitarClassifier-iOS My fork of that repo.
