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)