mindspore.ops.ImageSummary

查看源文件
class mindspore.ops.ImageSummary[源代码]

将图像保存到Summary文件。必须和SummaryRecord或SummaryCollector一起使用, Summary文件的保存路径由SummaryRecord或SummaryCollector指定。Summary文件可以通过MindInsight加载并展示, 关于MindInsight的详细信息请参考 MindInsight文档

在Ascend平台上的Graph模式下,可以通过设置环境变量 MS_DUMP_SLICE_SIZEMS_DUMP_WAIT_TIME 解决该算子在调用比较密集场景下算子执行失败的问题。

输入:
  • name (str) - 输入变量的名称,不能是空字符串。

  • value (Tensor) - 图像数据的值,Tensor的rank必须为4。

异常:
  • TypeError - 如果 name 不是str。

  • TypeError - 如果 value 不是Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.nn as nn
>>> from mindspore import ops
>>>
>>>
>>> class Net(nn.Cell):
...     def __init__(self):
...         super(Net, self).__init__()
...         self.summary = ops.ImageSummary()
...
...     def construct(self, x):
...         name = "image"
...         self.summary(name, x)
...         return x
...