mirror of
https://github.com/clearml/clearml
synced 2025-01-31 17:17:00 +00:00
Adds an tensorboardx example for add_video (creates GIFs in tensorboard)
This commit is contained in:
parent
1ab3710074
commit
4a2099b53c
28
examples/frameworks/tensorboardx/moviepy_tensorboardx.py
Normal file
28
examples/frameworks/tensorboardx/moviepy_tensorboardx.py
Normal 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()
|
Loading…
Reference in New Issue
Block a user