change name for jitter

pull/288/head
tink2123 2020-07-06 16:50:33 +08:00
parent 8d849cfa8c
commit f6d6249c93
1 changed files with 5 additions and 5 deletions

View File

@ -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:
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):
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: