/fabric-color-classification

Primary LanguageJupyter Notebook

Requirements

Install requirements by editing the command

pip install -r requirements

Data

Download the necessary data from this link.

Unzip it in the current folder

The .zip archive contains the fabric image dataset and the folder with the reference colors.

Code

  • Ensemble - CNN difference: it contains the code for creating the difference domain dataset, for performing the training of the three CNN difference models and the corresponding ensemble. Finally, it contains the performance measures.
  • CNN Difference and other models: it contains the code for creating the difference domain dataset, for performing the training of several CNN models available in Keras. Finally, it contains the performance measures.