提交 f6d6249c 编写于 作者: T tink2123

change name for jitter

上级 8d849cfa
...@@ -119,9 +119,9 @@ def blur(img): ...@@ -119,9 +119,9 @@ def blur(img):
return img return img
def doudong(img): def jitter(img):
""" """
doudong jitter
""" """
w, h, _ = img.shape w, h, _ = img.shape
if h > 10 and w > 10: if h > 10 and w > 10:
...@@ -198,7 +198,7 @@ class Config: ...@@ -198,7 +198,7 @@ class Config:
self.affine = False self.affine = False
self.reverse = True self.reverse = True
self.noise = True self.noise = True
self.dou = False self.jitter = True
self.blur = True self.blur = True
self.color = True self.color = True
...@@ -323,8 +323,8 @@ def warp(img, ang): ...@@ -323,8 +323,8 @@ def warp(img, ang):
tp = random.randint(1, 100) tp = random.randint(1, 100)
if tp >= 50: if tp >= 50:
new_img = cvtColor(new_img) new_img = cvtColor(new_img)
if config.dou: if config.jitter:
new_img = doudong(new_img) new_img = jitter(new_img)
if config.noise: if config.noise:
tp = random.randint(1, 100) tp = random.randint(1, 100)
if tp >= 50: if tp >= 50:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册