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.
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 email@example.com.
Want access to Two Tap's training data? Email firstname.lastname@example.org.