/ColorDetect

Image processing: Detect and identify different color objects in an image/video

Primary LanguagePythonMIT LicenseMIT

ColorDetect

ColorDetect
Documentation | Package

ColorDetect

Lint workflow PyPI version Python Package tests Downloads Documentation Status

ColorDetect works to recognize and identify different colors in an image or video.

Installation

pip install ColorDetect

Basic Usage

Images

from colordetect import ColorDetect


user_image = ColorDetect(<path_to_image>)
# return dictionary of color count. Do anything with this
user_image.get_color_count()

# write color count
user_image.write_color_count()
# optionally, write any text to the image
user_image.write_text(text="any text")

# save the image after using either of the options (write_color_count/write_text) or both
user_image.save_image(<storage_path>,<image_name>)

Resultant image is stored in the string storage_path of choice with the image_name which will default to the current location and out.jpg respectively by default.

Videos

from colordetect import VideoColor,col_share

user_video = VideoColor(<path_to_video>)
# return dictionary of color count. Do anything with this result
user_video.get_video_frames(progress=True)
# to order this rather long result and get only a specific number look up the `col_share` module

You can also get colors at a specific time and extract the frame at that given time.

Project Documentation

For further project documentation, visit ColorDetect's page

Contributions

Contributions are welcome. Do remember to take a look at the project contribution guidelines

Tests

To run tests:

pytest

Pre-commit

Pre-commit hooks are used to automate linting

  1. Install the git hook scripts

    pre-commit install
  2. (optional) Run against all the files

    pre-commit run --all-files

The installed pre-commit hooks will automatically ensure use of a consistent code format and style whenever one commits changes using git. For full documentation, view the pre-commit docs.

Hall of Code

To the amazing human beings and developers that made this possible.

contributors' avatars