Adds an tensorboardx example for add_video (creates GIFs in tensorboard)

This commit is contained in:
Tim Joseph 2021-05-31 13:33:56 +02:00 committed by Jake Henning
parent 1ab3710074
commit 4a2099b53c

View File

@ -0,0 +1,28 @@
# ClearML - Example of pytorch with tensorboardX add_video.
#
from __future__ import print_function
import torch
from tensorboardX import SummaryWriter
from clearml import Task
def main():
# Connecting ClearML with the current process,
# from here on everything is logged automatically
task = Task.init(project_name="examples", task_name="pytorch with tensorboardX")
writer = SummaryWriter("runs")
writer.add_text("TEXT", "This is some text", 0)
# Make a video that simply fades grey colors
video = (torch.sin(torch.arange(0, 1000) / 100) + 1) / 2 * 255
video = video.byte().view(1, -1, 1, 1, 1).expand(1, -1, 3, 64, 64)
writer.add_video("my_video", video, 0, fps=50)
if __name__ == "__main__":
main()