animation.py 文件源码

def train(self, training_agent, total_episode, save_movie=False):
        def animate_func(step):     
            #self.step = step       
            if step > total_episode:                 
                print("Finish training ")               
                #plt.close()
            else:
                training_agent(episode=step, visual=self)
                plt.pause(1.001) # makes the UI a little more responsive 

            if not self.ispause:
                return self._update()

        self.am = animation.FuncAnimation(self._fig, animate_func, total_episode, init_func=self._init, interval=16, repeat=False, blit=False)
        if save_movie:
            mywriter = animation.FFMpegWriter(fps=24, codec='libx264', extra_args=['-pix_fmt', 'yuv420p', '-profile:v', 'high', '-tune', 'animation', '-crf', '18'])
            self.am.save("./video.mp4", writer=mywriter)
        else:
            plt.show(block=True)