def __init__(self,
zone=(0.2, 0.8),
coverage=1.,
color='white',
noises=None,
name='Noise_Aug',
verbose=0,
stateless=True):
super().__init__(action=Action.SUBSTITUTE,
zone=zone,
coverage=coverage,
name=name,
device='cpu',
verbose=verbose,
stateless=stateless)
self.color = color
self.noises = noises
self.model = nma.Noise()
self.model.validate(color)
def get_model(cls, noise_factor):
return nma.Noise(noise_factor)
def get_model(self, nosie_factor):
return nma.Noise(nosie_factor)
def get_model(cls, zone, coverage, color, noises):
return nma.Noise(zone=zone, coverage=coverage, color=color, noises=noises)
