Perfect for profile picture processing for your website or batch work for ID cards, autocrop will output images centered around the biggest face detected.
Simple!
pip install autocrop
Autocrop can be used from the command line or directly from Python API.
Import the Cropper
class, set some parameters (optional), and start cropping.
The crop
method accepts filepaths or np.ndarray
, and returns Numpy arrays. These are easily handled with PIL or Matplotlib.
from PIL import Image
from autocrop import Cropper
cropper = Cropper()
# Get a Numpy array of the cropped image
cropped_array = cropper.crop('portrait.png')
# Save the cropped image with PIL if a face was detected:
if cropped_array:
cropped_image = Image.fromarray(cropped_array)
cropped_image.save('cropped.png')
Further examples and use cases are found in the accompanying Jupyter Notebook.
usage: [-h] [-o OUTPUT] [-i INPUT] [-w WIDTH] [-H HEIGHT] [-e EXTENSION] [-v]
Automatically crops faces from batches of pictures
optional arguments:
-h, --help
Show this help message and exit
-o, --output, -p, --path
Folder where cropped images will be placed.
Default: current working directory
-r, --reject
Folder where images without detected faces will be placed.
Default: same as output directory
-i, --input
Folder where images to crop are located.
Default: current working directory
-w, --width
Width of cropped files in px. Default=500
-H, --height
Height of cropped files in px. Default=500
--facePercent
Zoom factor. Percentage of face height to image height.
-e, --extension
Enter the image extension which to save at output.
Default: Your current image extension
-v, --version
Show program's version number and exit
- Crop every image in the
pics
folder, resize them to 400 px squares, and output them in thecrop
directory:autocrop -i pics -o crop -w 400 -H 400
.- Images where a face can't be detected will be left in
crop
.
- Same as above, but output the images with undetected faces to the
reject
directory:autocrop -i pics -o crop -r reject -w 400 -H 400
.
- Same as above but the image extension will be
png
:autocrop -i pics -o crop -w 400 -H 400 -e png
If no output folder is added, asks for confirmation and destructively crops images in-place.
You can use autocrop to detect faces in frames extracted from a video. A great way to perform the frame extraction step is with ffmpeg
:
mkdir frames faces
# Extract one frame per second
ffmpeg -i input.mp4 -filter:v fps=fps=1/60 frames/ffmpeg_%0d.bmp
# Crop faces as jpg
autocrop -i frames -o faces -e jpg
The following file types are supported:
- EPS files (
.eps
) - GIF files (
.gif
) (only the first frame of an animated GIF is used) - JPEG 2000 files (
.j2k
,.j2p
,.jp2
,.jpx
) - JPEG files (
.jpeg
,.jpg
,.jpe
) - LabEye IM files (
.im
) - macOS ICNS files (
.icns
) - Microsoft Paint bitmap files (
.msp
) - PCX files (
.pcx
) - Portable Network Graphics (
.png
) - Portable Pixmap files (
.pbm
,.pgm
,.ppm
) - SGI files (
.sgi
) - SPIDER files (
.spi
) - TGA files (
.tga
) - TIFF files (
.tif
,.tiff
) - WebP (
.webp
) - Windows bitmap files (
.bmp
,.dib
) - Windows ICO files (
.ico
) - X bitmap files (
.xbm
)
In some cases, you may wish the package directly, instead of through PyPI:
cd ~
git clone https://github.com/leblancfg/autocrop
cd autocrop
pip install .
Development of a conda-forge
package for the Anaconda Python distribution is currently stalled due to the complexity of setting up the workflow with OpenCV. Please leave feedback on issue #7 to see past attempts if you are insterested in helping out!
Best practice for your projects is of course to use virtual environments. At the very least, you will need to have pip installed.
Autocrop is currently being tested on:
- Python 3.6 to 3.10
- OS:
- Linux
- macOS
- Windows
Check out:
- http://docs.opencv.org/master/d7/d8b/tutorial_py_face_detection.html#gsc.tab=0
- http://docs.opencv.org/master/d5/daf/tutorial_py_histogram_equalization.html#gsc.tab=0
Adapted from:
Although autocrop is essentially a CLI wrapper around a single OpenCV function, it is actively developed. It has active users throughout the world.
If you would like to contribute, please consult the contribution docs.