mindspore.nn.CentralCrop

class mindspore.nn.CentralCrop(central_fraction)[源代码]

根据指定比例裁剪出图像的中心区域。

参数:

  • central_fraction (float) - 裁剪比例,必须是float,并且在范围(0.0, 1.0]内。

输入:

  • image (Tensor) - shape为 \((C, H, W)\) 的三维Tensor,或shape为 \((N,C,H,W)\) 的四维Tensor。

输出:

Tensor,基于输入的三维或四维的float Tensor。

异常:

  • TypeError - central_fraction 不是float。

  • ValueError - central_fraction 不在范围(0.0, 1.0]内。

支持平台:

Ascend GPU CPU

样例:

>>> net = nn.CentralCrop(central_fraction=0.5)
>>> image = Tensor(np.random.random((4, 3, 4, 4)), mindspore.float32)
>>> output = net(image)
>>> print(output.shape)
(4, 3, 2, 2)