Pytorch中使用tensorboard中如何添加torch.Tensor形式的图片add_image和add_images

发布时间:2021-12-04 19:00:30 作者:柒染
来源:亿速云 阅读:659

Pytorch中使用tensorboard中如何添加torch.Tensor形式的图片add_image和add_images,很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。

 add_image和add_images

代码:

from torch.utils.tensorboard import SummaryWriterimport numpy as np
img = np.zeros((3, 100, 100))img[0] = np.arange(0, 10000).reshape(100, 100) / 10000img[1] = 1 - np.arange(0, 10000).reshape(100, 100) / 10000img_HWC = np.zeros((100, 100, 3))img_HWC[:, :, 0] = np.arange(0, 10000).reshape(100, 100) / 10000img_HWC[:, :, 1] = 1 - np.arange(0, 10000).reshape(100, 100) / 10000writer = SummaryWriter()writer.add_image('my_image', img, 0)# If you have non-default dimension setting, set the dataformats argument.writer.add_image('my_image_HWC', img_HWC, 0, dataformats='HWC')writer.close()# tensorboard --logdir=runs

运行结果展示:
Pytorch中使用tensorboard中如何添加torch.Tensor形式的图片add_image和add_images

代码:

from torch.utils.tensorboard import SummaryWriterimport numpy as np

img_batch = np.zeros((16, 3, 100, 100))for i in range(16):img_batch[i, 0] = np.arange(0, 10000).reshape(100, 100) / 10000 / 16 * i
    img_batch[i, 1] = (1 - np.arange(0, 10000).reshape(100, 100) / 10000) / 16 * i

writer = SummaryWriter()writer.add_images('my_image_batch', img_batch, 0)writer.close()# tensorboard --logdir=runs

运行结果展示:Pytorch中使用tensorboard中如何添加torch.Tensor形式的图片add_image和add_images

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注亿速云行业资讯频道,感谢您对亿速云的支持。

推荐阅读:
  1. PyTorch中torch.tensor和torch.Tensor有什么区别
  2. PyTorch中TensorBoard如何使用

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

pytorch tensorboard torch.tensor

上一篇:pytorch创建data.DataLoader时对参数pin_memory的理解是什么

下一篇:PyTorch中梯度反向传播的注意点是什么

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》