/yolocrop

Use a YOLO model to just predict the bounding box of an object and crop it.

Primary LanguagePython

yolocrop

Use a YOLO model to just predict the bounding box of an object and crop it.

Designed for images where there is only a single object of interest.

Usage:

yolocrop WEIGHTS.pt IMAGE.jpg output/IMAGE-crop.jpg