No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
deep-learning-models
training
.gitignore
README.md
predict.py

README.md

Product Color Classification Engine

This is a snapshot of the product color classification engine used by Two Tap. It uses Inception v3 to try to identify the main color of a product image. The returned colors can be:

[ black, blue, brown, gold, green, orange, pink, purple, red, silver, white ]

How to use the it with the pre-trained weights.

  • Clone the repo.
  • Download the pretrained weights from here, unpack them, and place them in the models/ directory.
  • Run: python predict.py.

The models/ directory should look like below:

models-dir

How to train the model.

In order to train the model you have to use a directory structure like below. If you'd like access to our training data please email support@twotap.com.

ml-data-dir

Want access to Two Tap's training data? Email support@twotap.com.