No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Chris Navrides
Chris Navrides add README
Latest commit 4fa0cfb Oct 24, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
sample_run lower print threshold on run_script Oct 24, 2018
training_images initial commit Oct 24, 2018
LICENSE initial commit Oct 24, 2018
README.md add README Oct 24, 2018
retrain.py initial commit Oct 24, 2018

README.md

test.ai Classifier Builder

This is the classifier and training/running scripts for the test.ai elements classifier. This is used to build a model for the testai appium-classifier-plugin(https://github.com/testdotai/appium-classifier-plugin).

System setup

You will need to install Tensorflow(www.tensorflow.org), Google's open source AI framework.

pip install tensorflow

Build Classifier

To build the classifier from the provided images run the following.

python retrain.py --image_dir training_images/ --output_graph output/saved_model.pb --output_labels output/saved_model.pbtxt --how_many_training_steps 4000 --learning_rate 0.30 --testing_percentage 25 --validation_percentage 25 --eval_step_interval 50 --train_batch_size 2000 --test_batch_size -1 --validation_batch_size -1 --bottleneck_dir /tmp/bottleneck

Run Classifier

To use the classifier to classify images, run the script under sample_run/ directory.

python run_model.py cart.png