未验证 提交 78fdd43e 编写于 作者: W wangxinxin08 提交者: GitHub

fix gridmask op in dygraph and modify some comments (#2560)

上级 5152c739
...@@ -45,7 +45,7 @@ class GridMask(object): ...@@ -45,7 +45,7 @@ class GridMask(object):
self.prob = self.st_prob * min(1, 1.0 * curr_iter / self.upper_iter) self.prob = self.st_prob * min(1, 1.0 * curr_iter / self.upper_iter)
if np.random.rand() > self.prob: if np.random.rand() > self.prob:
return x return x
_, h, w = x.shape h, w, _ = x.shape
hh = int(1.5 * h) hh = int(1.5 * h)
ww = int(1.5 * w) ww = int(1.5 * w)
d = np.random.randint(2, h) d = np.random.randint(2, h)
......
...@@ -165,7 +165,7 @@ class Permute(BaseOperator): ...@@ -165,7 +165,7 @@ class Permute(BaseOperator):
@register_op @register_op
class Lighting(BaseOperator): class Lighting(BaseOperator):
""" """
Lighting the imagen by eigenvalues and eigenvectors Lighting the image by eigenvalues and eigenvectors
Args: Args:
eigval (list): eigenvalues eigval (list): eigenvalues
eigvec (list): eigenvectors eigvec (list): eigenvectors
...@@ -1774,12 +1774,13 @@ class Pad(BaseOperator): ...@@ -1774,12 +1774,13 @@ class Pad(BaseOperator):
offsets=None, offsets=None,
fill_value=(127.5, 127.5, 127.5)): fill_value=(127.5, 127.5, 127.5)):
""" """
Pad image to a specified size or multiple of size_divisor. random target_size and interpolation method Pad image to a specified size or multiple of size_divisor.
Args: Args:
size (int, Sequence): image target size, if None, pad to multiple of size_divisor, default None size (int, Sequence): image target size, if None, pad to multiple of size_divisor, default None
size_divisor (int): size divisor, default 32 size_divisor (int): size divisor, default 32
pad_mode (int): pad mode, currently only supports four modes [-1, 0, 1, 2]. if -1, use specified offsets pad_mode (int): pad mode, currently only supports four modes [-1, 0, 1, 2]. if -1, use specified offsets
if 0, only pad to right and bottom. if 1, pad according to center. if 2, only pad left and top if 0, only pad to right and bottom. if 1, pad according to center. if 2, only pad left and top
offsets (list): [offset_x, offset_y], specify offset while padding, only supported pad_mode=-1
fill_value (bool): rgb value of pad area, default (127.5, 127.5, 127.5) fill_value (bool): rgb value of pad area, default (127.5, 127.5, 127.5)
""" """
super(Pad, self).__init__() super(Pad, self).__init__()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册