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.
Failed to load latest commit information.

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

The models/ directory should look like below:


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


Want access to Two Tap's training data? Email