def _get_augment_params(self, img): h, w = img.shape[:2] scale = self.size * 1.0 / min(h, w) if h < w: newh, neww = self.size, scale * w else: newh, neww = scale * h, self.size if max(newh, neww) > self.max_size: scale = self.max_size * 1.0 / max(newh, neww) newh = newh * scale neww = neww * scale neww = int(neww + 0.5) newh = int(newh + 0.5) return transform.ResizeTransform(h, w, newh, neww, self.interp)
def _get_augment_params(self, img): h, w = img.shape[:2] return transform.ResizeTransform(h, w, self.size[0], self.size[1], self.interp)