(self, inputs, target)
| 416 | self.color = np.exp(np.random.normal(0, color*schedule_coeff,3)) |
| 417 | |
| 418 | def __call__(self, inputs, target): |
| 419 | inputs[1] = self.chrom_aug(inputs[1]) |
| 420 | # noise |
| 421 | inputs[0]+=np.random.normal(0, self.noise, inputs[0].shape) |
| 422 | inputs[1]+=np.random.normal(0, self.noise, inputs[0].shape) |
| 423 | return inputs,target |
| 424 | |
| 425 | def chrom_aug(self, rgb): |
| 426 | # color change |