媒体数据

可视化神经网络中的样本数据,如图片,音频,文本等。

图像

1.单张图片

summarywriter.add_image(tag: str,
                        tensor: numpy_compatible,
                        step: Optional[int] = None):
    """
        添加单张图像数据到日志
    Args:
        tag: 字符串,图像的标识
        tensor: 数组,图像数据,'uint8' 或 'float' 类型的数据,大小为(H,W) 或 (H,W,C), 其中C为1,2,3,4
        step: 整数,可选参数,记录数据的step
    """

2.多张图片

summarywriter.add_images(tag: str,
                         tensors: numpy_compatible,
                         step: Optional[int] = None):
    """
        添加多个图像数据到日志
    Args:
        tag: 字符串, 图像的标识, 导出过程中将自动转化为tag_1, tag_2, ... , tag_k
        tensors: 数组类型,图像数据,'uint8' 或 'float' 类型的数据,大小为(K,H,W) 或 (K,H,W,C), 其中C为1,3,4
        step: 可选参数,记录数据的step
    """

文本

summarywriter.add_text(tag: str,
                       text_string: str,
                       step: Optional[int] = None):
    """
        添加文本数据到日志
    Args:
        tag: 字符串,文本的标识
        text_string: 字符串,文本字符串
        step: 整数,可选参数,记录数据的step
    """

音频

summarywriter.add_audio(tag: str,
                        audio_tensor: numpy_compatible,
                        step: Optional[int],
                        sample_rate: Optional[int] = 44100):
    """
        添加音频数据到日志
    Args:
        tag: 字符串,音频的标识
        audio_tensor: 数组,音频数据,大小为(L, C), 其中L为音频帧的长度,C为通道,通常C=1,2
        step: 整数,可选参数,记录数据的step
        sample_rate: 整数,采样率 Hz
    """