You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
imaginAIry/tests/test_enhancers/test_facecrop.py

13 lines
376 B
Python

import logging
from imaginairy.enhancers.facecrop import generate_face_crops
from imaginairy.schema import LazyLoadingImage
from tests import TESTS_FOLDER
logger = logging.getLogger(__name__)
def test_facecrop():
img = LazyLoadingImage(filepath=f"{TESTS_FOLDER}/data/bench2.png")
generate_face_crops((50, 50, 150, 150), max_width=img.width, max_height=img.height)