colour-science / colour-checker-detection

Colour checker detection with Python
https://www.colour-science.org/
BSD 3-Clause "New" or "Revised" License
221 stars 31 forks source link

PR: Machine Learning Inference #75

Closed KelSolaar closed 9 months ago

KelSolaar commented 9 months ago

Summary

This PR implements support for machine learning inference to detect the colour checker.

The model is published on HuggingFace, and was trained on a purposely constructed dataset.

The model has only been trained on ColorChecker Classic 24 images and will not work with ColorChecker Nano or ColorChecker SG images.

Inference is performed by a script licensed under the terms of the GNU Affero General Public License v3.0 as it uses the Ultralytics YOLOv8 API which is incompatible with the BSD-3-Clause.

image

image

image

The segmentation API has been overhauled to provide a consistent interface across the two methods. It has also been improved following discussions in #53, #73, and #74.

Preflight

Code Style and Quality

Documentation

KelSolaar commented 9 months ago

Some extreme examples:

image image image image image image image image