Pytorch中使用tensorboard中如何添加torch.Tensor形式的图片add_image和add_images
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
运行结果展示:
代码:
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
当前网址:http://myzitong.com/article/pgdpph.html