mxnet.gluon.data.vision.transforms.ToTensor的作用

最新推荐文章于 2022-11-10 12:33:22 发布

原创最新推荐文章于 2022-11-10 12:33:22 发布 · 1.4k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#ToTensor #mxnet

深度学习专栏收录该内容

23 篇文章

订阅专栏

本文详细介绍了MXNet库中ToTensor转换器的功能和使用方法，该转换器用于将图像数据从[0,255]的整数范围转换到[0,1]的浮点数范围，并调整其形状从(HxWxC)到(CxHxW)，适用于深度学习模型的输入。通过示例展示了如何使用ToTensor进行图像预处理。

mxnet.gluon.data.vision.transforms.ToTensor的作用：

将图像或成批图像转换为张量NDArray。
将[0,255]范围内的图像（NDArray of shape (H x W x C)）转换为[0,1]范围内的float32张量（NDArray of shape (C x H x W)）。
例子：

from mxnet.gluon import data as gdata
from mxnet import nd
import numpy as np
transformer = gdata.vision.transforms.ToTensor()
#从均匀分布中随机抽取样本,样品均匀分布在半开区间[低，高](含低，不含高)。
image = nd.random.uniform(0, 255, (4, 2, 3)).astype(dtype=np.uint8)
print(image)
print(transformer(image))

运行结果：
在这里插入图片描述

其中，均匀分布uniform用法如下：