Color Thief

A module for detecting shapes using OpenCV.

Installation

pip install shape-detection-jdla

Overview

Square detection rules:

• There must be four corners
• All four lines must be the same length
• All four corners must be 90°
• Lines AB and CD must be horizontal lines
• Lines AC and BC must be vertical lines
• The contour must be concave

Example

from shape_detection.square import Square
import numpy as np

try:
    contour = np.array([[[368, 160]], [[391, 163]],
                        [[384, 200]], [[361, 194]]])

    square = Square.is_square(contour)

    print(square)

except Exception as e:
    print(e)

Links

• github