from pybug.io import auto_import
from pybug.image import RGBImage, MaskedNDImage
images = auto_import('/Users/joan/PhD/DataBases/lfpw/trainset/*.png')
images = [i.as_greyscale() if type(i) is RGBImage else i for i in images]
new_images = []
for i in images:
img = MaskedNDImage(i.pixels)
img.landmarks = i.landmarks
new_images.append(img)
images = new_images
del new_images
# <codecell>
from pybug.activeappearancemodel import aam_builder, AAM
from pybug.transform.tps import TPS
from pybug.landmark import ibug_68_trimesh
options = {'triangulation': {'function': ibug_68_trimesh, 'label': 'ibug_68_trimesh'},
'features': {'type': None, 'options': {'kwargs': None}},
'max_shape_components': 25,
'max_appearance_components': 250}
aam = aam_builder(images, **options)
import numpy as np
# <codecell>
