18 lines
507 B
Python
18 lines
507 B
Python
from pathlib import Path
|
|
|
|
from skimage.io import imread, imsave
|
|
|
|
from falign.landmarks import read_image_and_get_landmarks
|
|
from falign.align import align
|
|
|
|
|
|
def test_align():
|
|
dir_gallery = Path(__file__).parent.parent.parent / "gallery"
|
|
path_image = dir_gallery / "original.jpg"
|
|
|
|
image, landmarks = read_image_and_get_landmarks(path_image)
|
|
aligned = align(image, landmarks, height=256, width=256)
|
|
assert aligned.shape == (256, 256, 3)
|
|
|
|
imsave(dir_gallery / "test_align.jpg", aligned)
|