DlibWrapper

class menpofit.dlib.DlibWrapper(model)[source]

Bases: object

Wrapper class for fitting a pre-trained ERT model. Pre-trained models are provided by the official DLib package (http://dlib.net/).

Parameters

model (Path or str) – Path to the pre-trained model.

fit_from_bb(image, bounding_box, gt_shape=None)[source]

Fits the model to an image given an initial bounding box.

Parameters

  • image (menpo.image.Image or subclass) – The image to be fitted.

  • bounding_box (menpo.shape.PointDirectedGraph) – The initial bounding box.

  • gt_shape (menpo.shape.PointCloud) – The ground truth shape associated to the image.

Returns

fitting_result (Result) – The result of the fitting procedure.

fit_from_shape(image, initial_shape, gt_shape=None)[source]

Fits the model to an image. Note that it is not possible to initialise the fitting process from a shape. Thus, this method raises a warning and calls fit_from_bb with the bounding box of the provided initial_shape.

Parameters

  • image (menpo.image.Image or subclass) – The image to be fitted.

  • initial_shape (menpo.shape.PointCloud) – The initial shape estimate from which the fitting procedure will start. Note that the shape won’t actually be used, only its bounding box.

  • gt_shape (menpo.shape.PointCloud) – The ground truth shape associated to the image.

Returns

fitting_result (Result) – The result of the fitting procedure.