/cropimage

A simple toolkit for detecting and cropping main body from pictures. Support face and saliency detection.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

cropimage

Open in Colab PyPI version

cropimage is a simple toolkit for image cropping. Different from other projects that mainly tackle face region, we extend to all kinds of images such as landscape. We support horizontal and portrait size and generate a square output.

Installation

pip install cropimage

Get Started

from cropimage import Cropper

cropper = Cropper()

# Get a Numpy array of the cropped image
# Set completeness to be False if you expect the 'person' to be complete rather than 'face'
# Set target_size to be a tuple (size, size), only square output is supported now,
result = cropper.crop('./images/input.jpg')

# Save the cropped image
cv2.imwrite('cropped.jpg', result)

# Just get the cropping region
result = cropper.crop('./images/input.jpg',region_only=True)

More Results

Contributing

If you find any issue of this project, feel free to open an issue or contribute!