BEAT - tutorial/gmm_projection/1

This algorithm is a legacy one. The API has changed since its implementation. New versions and forks will need to be updated.

Endpoint Groups 2

Algorithms have at least one input and one output. All algorithm endpoints are organized in groups. Groups are used by the platform to indicate which inputs and outputs are synchronized together. The first group is automatically synchronized with the channel defined by the block in which the algorithm is deployed.

Group: main

Endpoint Name	Data Format	Nature
features	system/array_2d_floats/1	Input
id	system/uint64/1	Input
model	tutorial/gmm/1	Output

Unnamed group

Endpoint Name	Data Format	Nature
ubm	tutorial/gmm/1	Input

import bob
import numpy
from bob.machine import GMMMachine

def gmm_from_data(data):
    """Unmangles a bob.machine.GMMMachine from a BEAT Data object"""

dim_c, dim_d = data.means.shape
    gmm = GMMMachine(dim_c, dim_d)
    gmm.weights = data.weights
    gmm.means = data.means
    gmm.variances = data.variances
    gmm.variance_thresholds = data.variance_thresholds
    return gmm

class Algorithm:

def __init__(self):
        self.ubm      = None
        self.features = []

def process(self, inputs, outputs):

# retrieve the UBM once
        if self.ubm is None:
            inputs['ubm'].next()
            self.ubm = gmm_from_data(inputs['ubm'].data)

# collect all the features for the current template
        self.features.append(inputs["features"].data.value)

# adapts the UBM GMM for the template (when all the features have been collected)
        if inputs["id"].isDataUnitDone():
            model = bob.machine.GMMMachine(self.ubm)

trainer = bob.trainer.MAP_GMMTrainer()
            trainer.max_iterations = 1
            trainer.set_prior_gmm(self.ubm)
            trainer.train(model, numpy.vstack(self.features))

# outputs data
            outputs["model"].write({
                'weights':              model.weights,
                'means':                model.means,
                'variances':            model.variances,
                'variance_thresholds':  model.variance_thresholds,
            })

self.features = []

return True

The code for this algorithm in Python
The ruler at 80 columns indicate suggested POSIX line breaks (for readability).
The editor will automatically enlarge to accomodate the entirety of your input
Use keyboard shortcuts for search/replace and faster editing. For example, use Ctrl-F (PC) or Cmd-F (Mac) to search through this box

Could not find any documentation for this object.

No experiments are using this algorithm.

This algorithm was never executed.

algorithms tutorial gmm_projection 1

Endpoint Groups 2

Group: main

Unnamed group

algorithms

tutorial

gmm_projection

1