Comment animer une séquence chronologique de parcelles de matplotlib

Je veux tracer une séquence de .les images png en matplotlib. L'objectif est de tracer rapidement pour simuler l'effet d'un film, mais j'ai d'autres raisons de vouloir éviter de créer réellement un .fichier avi ou de l'enregistrement matplotlib chiffres et l'affichage dans l'ordre, à l'extérieur de Python.

Je suis spécialement essayez d'afficher les fichiers d'image dans la séquence à l'intérieur d'une boucle en Python. En supposant que j'ai importé matplotlib correctement, et j'ai mes propres fonctions 'new_image()' et 'new_rect()', voici un exemple de code qui ne fonctionne pas en raison de l'effet de blocage de la méthode show() de l'appel à la GUI mainloop:

 for index in index_list:
     img = new_image(index)
     rect = new_rect(index)

     plt.imshow(img)
     plt.gca().add_patch(rect)
     plt.show()

     #I also tried pausing briefly and then closing, but this doesn't
     #get executed due to the GUI mainloop from show()
     time.sleep(0.25)
     plt.close()

Le code ci-dessus fonctionne pour n'afficher que la première image, mais ensuite, le programme se bloque et m'attend pour fermer manuellement la résultante de la figure de la fenêtre. Une fois que je ne la fermez, puis le programme se bloque et ne pas re-terrain avec les nouvelles données de l'image. Que dois-je faire? Note aussi que j'ai essayé de remplacer le plt.show() commande avec un plt.draw() de commande, puis en ajoutant le plt.show() en dehors de la boucle for. Cela ne veut pas afficher quoi que ce soit et se contente d'.

source d'informationauteur ely

  1. 8

    Basé sur http://matplotlib.sourceforge.net/examples/animation/simple_anim_tkagg.html:

    import time
import numpy as np
import matplotlib
matplotlib.use('TkAgg') # do this before importing pylab

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)

def animate():
    tstart = time.time()                   # for profiling
    data=np.random.randn(10,10)
    im=plt.imshow(data)

    for i in np.arange(1,200):
        data=np.random.randn(10,10)
        im.set_data(data)
        fig.canvas.draw()                         # redraw the canvas
    print 'FPS:' , 200/(time.time()-tstart)

win = fig.canvas.manager.window
fig.canvas.manager.window.after(100, animate)
plt.show()

    plt.imshow peut accepter un float array, array uint8, ou une image de PIL.
    Donc, si vous avez un répertoire de fichiers PNG, vous pouvez les ouvrir comme PIL images et de les animer comme ceci:

    import matplotlib
matplotlib.use('TkAgg') # do this before importing pylab
import matplotlib.pyplot as plt
import Image
import glob

fig = plt.figure()
ax = fig.add_subplot(111)

def animate():
    filenames=sorted(glob.glob('*.png'))
    im=plt.imshow(Image.open(filenames[0]))
    for filename in filenames[1:]:
        image=Image.open(filename)
        im.set_data(image)
        fig.canvas.draw() 

win = fig.canvas.manager.window
fig.canvas.manager.window.after(100, animate)
plt.show()
  2. 6

    La meilleure façon que j'ai trouvé pour cet été avec la commande pylab.ion() après l'import pylab.

    Voici un script qui utilise la show()mais qui affiche les différentes parcelles à chaque fois pylab.draw() est appelé, et qui laisse l'intrigue windows montrant indéfiniment. Il utilise de simples logique d'entrée pour décider du moment de fermer les chiffres (parce que l'utilisation show() signifie pylab ne sera pas le traiter clique sur le windows bouton x), mais qui devrait être simple à ajouter à votre interface utilisateur graphique, comme un bouton ou un champ de texte.

    import numpy as np
import pylab
pylab.ion()

def get_fig(fig_num, some_data, some_labels):

    fig = pylab.figure(fig_num,figsize=(8,8),frameon=False)
    ax = fig.add_subplot(111)
    ax.set_ylim([0.1,0.8]); ax.set_xlim([0.1, 0.8]);
    ax.set_title("Quarterly Stapler Thefts")
    ax.pie(some_data, labels=some_labels, autopct='%1.1f%%', shadow=True);
    return fig

my_labels = ("You", "Me", "Some guy", "Bob")

# To ensure first plot is always made.
do_plot = 1; num_plots = 0;

while do_plot:
    num_plots = num_plots + 1;
    data = np.random.rand(1,4).tolist()[0]

    fig = get_fig(num_plots,data,my_labels)
    fig.canvas.draw()
    pylab.draw()

    print "Close any of the previous plots? If yes, enter its number, otherwise enter 0..."
    close_plot = raw_input()

    if int(close_plot) > 0:
        pylab.close(int(close_plot))

    print "Create another random plot? 1 for yes; 0 for no."
    do_plot = raw_input();

    # Don't allow plots to go over 10.
    if num_plots > 10:
        do_plot = 0

pylab.show()

    Par la modification de la logique de base ici, je peux l'avoir à fermer les fenêtres et de la parcelle images consécutives pour simuler la lecture d'un film, ou je peux les maintenir clavier de contrôle sur la façon dont il suit à travers le film.

    Remarque: Ce qui a fonctionné pour moi sur toutes les plateformes et semble strictement supérieure à la fenêtre de la toile gestionnaire de l'approche ci-dessus, et ne nécessite pas l' 'TkAgg' option.

  3. 0

    J'ai mis en œuvre une pratique de script qui vient convient à votre besoin. Essayer ici

    Ci-dessous est un exemple qui montrent des images de concert avec sa boite:

    import os
import glob
from scipy.misc import imread
from matplotlib.pyplot import Rectangle

video_dir = 'YOUR-VIDEO-DIRECTORY'

img_files = glob.glob(os.path.join(video_dir, '*.jpg'))
box_files = glob.glob(os.path.join(video_dir, '*.txt'))

def redraw_fn(f, axes):
    img = imread(img_files[f])
    box = bbread(box_files[f])  # Define your own bounding box reading utility
    x, y, w, h = box
    if not redraw_fn.initialized:
        im = axes.imshow(img, animated=True)
        bb = Rectangle((x, y), w, h,
                       fill=False,  # remove background
                       edgecolor="red")
        axes.add_patch(bb)
        redraw_fn.im = im
        redraw_fn.bb = bb
        redraw_fn.initialized = True
    else:
        redraw_fn.im.set_array(img)
        redraw_fn.bb.set_xy((x, y))
        redraw_fn.bb.set_width(w)
        redraw_fn.bb.set_height(h)
redraw_fn.initialized = False

videofig(len(img_files), redraw_fn, play_fps=30)