当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.gluon.data.vision.transforms.Resize用法及代码示例


用法:

class mxnet.gluon.data.vision.transforms.Resize(size, keep_ratio=False, interpolation=1)

参数

  • size(int or tuple of (W, H)) - 输出图像的大小。
  • keep_ratio(bool) - 是否调整短边或两边的大小size, 如果 size 作为整数给出。
  • interpolation(int) - 调整大小的插值方法。默认情况下使用双线性插值。有关可用选项,请参阅 OpenCV 的调整大小函数。请注意,GPU 上的 Resize 使用 contrib.bilinearResize2D 运算符,它仅支持双线性插值(1)。

基础:mxnet.gluon.block.HybridBlock

将图像或一批图像 NDArray 调整为给定大小。应该在 mxnet.gluon.data.vision.transforms.ToTensor 之前应用。

输入:
  • data:输入具有 (H x W x C) 或 (N x H x W x C) 形状的张量。
输出:
  • out:输出具有 (H x W x C) 或 (N x H x W x C) 形状的张量。

例子

>>> transformer = vision.transforms.Resize(size=(1000, 500))
>>> image = mx.nd.random.uniform(0, 255, (224, 224, 3)).astype(dtype=np.uint8)
>>> transformer(image)
<NDArray 500x1000x3 @cpu(0)>
>>> image = mx.nd.random.uniform(0, 255, (3, 224, 224, 3)).astype(dtype=np.uint8)
>>> transformer(image)
<NDArray 3x500x1000x3 @cpu(0)>

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.gluon.data.vision.transforms.Resize。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。