random_walk.py

#!/usr/bin/env python
"""
Random walk face generator

Dependends on:
    * numpy 1.7
    * matplotlib
"""
import numpy
from matplotlib import pyplot
from morpher import Morpher


if __name__ == "__main__":
    # Initialize face generator
    morpher = Morpher()

    # Make sure I/O doesn't hang when displaying the image
    pyplot.ion()

    # Build visualization window
    fig = pyplot.figure(figsize=(1, 1), dpi=300)
    im = pyplot.imshow(X=morpher.generate_face(),
                       interpolation='nearest',
                       cmap='gray')
    pyplot.axis('off')
    pyplot.show()

    # Program loop
    print "Press CTRL+C to stop"
    while True:
        Z = morpher.get_Z()
        dimension = morpher.index_mapping[numpy.random.choice(29)]
        delta = 0.5 * numpy.random.choice([-1, 1])
        # delta = numpy.random.normal(scale=0.5)
        Z[dimension] = Z[dimension] + delta
        morpher.set_Z(Z)
        im.set_array(morpher.generate_face())
        pyplot.draw()