mirror of
https://github.com/clearml/clearml
synced 2025-01-31 09:07:00 +00:00
29 lines
758 B
Python
29 lines
758 B
Python
# 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 video 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()
|