Create and Deploy a REST API Extracting Predominant Colors from Images

Using unsupervised machine learning, FastAPI and Docker

Image by author.

Table of contents

1. Problem statement
2. Extract colors from images
3. Project structure
4. Code
5. Deploy the Docker container
6. Let’s try it!
7. API documentation
8. Conclusions
9. License disclaimer

1. Problem statement

Let us imagine a control room of a manufacturing facility, where the fabricated products need to be sorted automatically. For instance, based on their color, goods may be redirected to different branches of a roller conveyor for further processing or packaging.

Otherwise, we can also imagine an online retailer trying to enhance the user experience by adding a search-by-color functionality. Customers may more easily find a clothing item from a particular color, thus simplifying their access to products of interest.

Or, just like the author, you can picture yourself as an IT consultant implementing a simple, fast and reusable tool to generate color palettes for presentations, charts and…