未验证 提交 08158e93 编写于 作者: J JYChen 提交者: GitHub

update RandomErasing document (#42627)

上级 66a10f36
...@@ -714,9 +714,33 @@ def erase(img, i, j, h, w, v, inplace=False): ...@@ -714,9 +714,33 @@ def erase(img, i, j, h, w, v, inplace=False):
import paddle import paddle
fake_img = paddle.randn((3, 10, 10)).astype(paddle.float32) fake_img = paddle.randn((3, 2, 4)).astype(paddle.float32)
print(fake_img)
#Tensor(shape=[3, 2, 4], dtype=float32, place=Place(gpu:0), stop_gradient=True,
# [[[ 0.02169025, -0.97859967, -1.39175487, -1.07478464],
# [ 0.20654772, 1.74624777, 0.32268861, -0.13857445]],
#
# [[-0.14993843, 1.10793507, -0.40056887, -1.94395220],
# [ 0.41686651, 0.44551995, -0.09356714, -0.60898107]],
#
# [[-0.24998808, -1.47699273, -0.88838995, 0.42629015],
# [ 0.56948012, -0.96200180, 0.53355658, 3.20450878]]])
values = paddle.zeros((1,1,1), dtype=paddle.float32) values = paddle.zeros((1,1,1), dtype=paddle.float32)
result = paddle.vision.transforms.erase(fake_img, 4, 4, 3, 3, values) result = paddle.vision.transforms.erase(fake_img, 0, 1, 1, 2, values)
print(result)
#Tensor(shape=[3, 2, 4], dtype=float32, place=Place(gpu:0), stop_gradient=True,
# [[[ 0.02169025, 0. , 0. , -1.07478464],
# [ 0.20654772, 1.74624777, 0.32268861, -0.13857445]],
#
# [[-0.14993843, 0. , 0. , -1.94395220],
# [ 0.41686651, 0.44551995, -0.09356714, -0.60898107]],
#
# [[-0.24998808, 0. , 0. , 0.42629015],
# [ 0.56948012, -0.96200180, 0.53355658, 3.20450878]]])
""" """
if _is_tensor_image(img): if _is_tensor_image(img):
......
...@@ -1378,6 +1378,8 @@ class RandomErasing(BaseTransform): ...@@ -1378,6 +1378,8 @@ class RandomErasing(BaseTransform):
fake_img = paddle.randn((3, 10, 10)).astype(paddle.float32) fake_img = paddle.randn((3, 10, 10)).astype(paddle.float32)
transform = paddle.vision.transforms.RandomErasing() transform = paddle.vision.transforms.RandomErasing()
result = transform(fake_img) result = transform(fake_img)
print(result)
""" """
def __init__(self, def __init__(self,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册