smarts.env.wrappers.gif_recorder module

class smarts.env.wrappers.gif_recorder.GifRecorder(video_name_folder: str, env: gymnasium.Env)

Uses images (such as from gym’s "rgb_array") to create a gif file.

capture_frame(step_num: int, image: ndarray)

Create image according to the "rgb_array" and store it with step number in the destination folder

close_recorder()

close the recorder by deleting the image folder.

generate_gif()

Use the images in the same folder to create a gif file.