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
