def generate_patches(scaled_imgs, constants): patch_size = constants.PATCH_SIZE std_dev_threshold = constants.STD_DEV_THRESHOLD patches = [] for k, sc in enumerate(scaled_imgs): img_patches = [] for i in range(sc.shape[0] - patch_size): for j in range(sc.shape[1] - patch_size): raw_patch = sc[i:i + patch_size, j:j + patch_size, :] patch = Patch( raw_patch=raw_patch, patch_size=patch_size, ) if patch.std_dev > std_dev_threshold: patch.store(sc, [i, j]) img_patches.append(patch) patches.append(img_patches) return patches
def generate_patches(scaled_imgs, constants, all_patches): """Generate patches for all scaled images.""" patch_size = constants.PATCH_SIZE step = 1 if all_patches else 2 patches = [] for k, sc in enumerate(scaled_imgs): img_patches = [] for i in range(0, sc.shape[0] - patch_size, step): for j in range(0, sc.shape[1] - patch_size, step): raw_patch = sc[i:i + patch_size, j:j + patch_size, :] patch = Patch( raw_patch=raw_patch, patch_size=patch_size, ) patch.store(sc, [i, j]) img_patches.append(patch) patches.append(img_patches) return patches