def preprocess_image(self, image_path, mask_path): image = load_image(image_path) mask = load_image(mask_path) image = resize(image, self.size) mask = resize(mask, self.size) mask = (mask == 1) * 1.0 if self.flip: if self.prng.choice([True, False]): image = np.flip(image, axis=1) mask = np.flip(mask, axis=1) return image * mask
def preprocess_image(self, image_path: str, mask_path: str) -> np.ndarray: image = load_image(image_path) mask = load_image(mask_path) mask = mask == self.mask_label if self.invert_mask: mask = np.logical_not(mask) image = image * 1.0 * mask image = resize(image, self.size) if self.flip: if self.prng.choice([True, False]): image = np.flip(image, axis=1) return image
def preprocess_image(self, image_path): image = load_image(image_path) image = resize(image, self.size) if self.flip: if self.prng.choice([True, False]): image = np.flip(image, axis=1) return image
def preprocess_image(self, image_path): image = load_image(image_path) image = resize(image, self.size) image = center_crop(image) return image