Method
__init__
(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5, mode=0, prob = 1.)
Source from the content-addressed store, hash-verified
| 6 | |
| 7 | class Grid(object): |
| 8 | def __init__(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5, mode=0, prob = 1.): |
| 9 | self.use_h = use_h |
| 10 | self.use_w = use_w |
| 11 | self.rotate = rotate |
| 12 | self.offset = offset |
| 13 | self.ratio = ratio |
| 14 | self.mode=mode |
| 15 | self.st_prob = prob |
| 16 | self.prob = prob |
| 17 | |
| 18 | def set_prob(self, epoch, max_epoch): |
| 19 | self.prob = self.st_prob * epoch / max_epoch |
Tested by
no test coverage detected