tensorboardx_summarywriter_log_scalars_images_text_histograms.py

1import torch
2import numpy as np
3from tensorboardX import SummaryWriter
4
5writer = SummaryWriter()
6
7for n_iter in range(100):
8    writer.add_scalar('data/scalar_systemtime', 10 + n_iter, n_iter)
9    writer.add_scalar('data/scalar_sum', 10 + n_iter, n_iter)
10    writer.add_scalars('data/scalar_group', {'xansin': n_iter, 'xacosn': n_iter}, n_iter)
11    
12    # Exporting a dummy image
13    writer.add_image('Image', np.ones((3, 64, 64)), n_iter)
14    
15    # Exporting text
16    writer.add_text('Text', 'text logged at step:' + str(n_iter), n_iter)
17
18    # Exporting a histogram
19    writer.add_histogram('hist', np.random.randn(100), n_iter)
20
21    # Exporting a precision-recall curve
22    writer.add_pr_curve('xdetec', np.random.randint(2, size=100), np.random.rand(100), n_iter)
23
24writer.close()