def draw_small_overview():
ia.seed(42)
image = ia.quokka(size=0.2)
heatmap = ia.quokka_heatmap(size=0.4)
segmap = ia.quokka_segmentation_map(size=0.2)
kps = ia.quokka_keypoints(size=0.2)
bbs = ia.quokka_bounding_boxes(size=0.2)
batch = ia.Batch(images=[image],
heatmaps=[heatmap.invert()],
segmentation_maps=[segmap],
keypoints=[kps],
bounding_boxes=[bbs])
augs = []
augs.append(("noop", iaa.Noop()))
augs.append(("non_geometric",
iaa.Sequential([
iaa.AdditiveGaussianNoise(scale=(0, 20)),
iaa.ContrastNormalization(1.2),
iaa.Sharpen(alpha=1.0, lightness=1.5)
])))
augs.append(("affine",
iaa.Affine(rotate=0,
translate_percent={"x": 0.1},
scale=1.3,
mode="constant",
cval=25)))
augs.append(("cropandpad",
iaa.CropAndPad(percent=(-0.05, 0.2, -0.05, -0.2),
pad_mode="maximum")))
augs.append(("fliplr_perspective",
iaa.Sequential(
[iaa.Fliplr(1.0),
iaa.PerspectiveTransform(scale=0.15)])))
for name, aug in augs:
result = list(aug.augment_batches([batch]))[0]
image_aug = result.images_aug[0]
image_aug_heatmap = result.heatmaps_aug[0].draw(cmap=None)[0]
image_aug_segmap = result.segmentation_maps_aug[0].draw_on_image(
image_aug, alpha=0.8)
image_aug_kps = result.keypoints_aug[0].draw_on_image(
image_aug, color=[0, 255, 0], size=7)
image_aug_bbs = result.bounding_boxes_aug[0].cut_out_of_image(
).draw_on_image(image_aug, thickness=3)
imageio.imwrite(
os.path.join(IMAGES_DIR, "small_overview",
"%s_image.jpg" % (name, )), image_aug)
imageio.imwrite(
os.path.join(IMAGES_DIR, "small_overview",
"%s_heatmap.jpg" % (name, )), image_aug_heatmap)
imageio.imwrite(
os.path.join(IMAGES_DIR, "small_overview",
"%s_segmap.jpg" % (name, )), image_aug_segmap)
imageio.imwrite(
os.path.join(IMAGES_DIR, "small_overview",
"%s_kps.jpg" % (name, )), image_aug_kps)
imageio.imwrite(
os.path.join(IMAGES_DIR, "small_overview",
"%s_bbs.jpg" % (name, )), image_aug_bbs)
def main():
image = ia.quokka_square((200, 200))
kpsoi = ia.quokka_keypoints((200, 200), extract="square")
aug = iaa.Jigsaw(10, 10)
images_aug, kpsois_aug = aug(images=[image] * 16,
keypoints=[kpsoi] * 16)
images_show = [kpsoi_aug.draw_on_image(image_aug)
for image_aug, kpsoi_aug in zip(images_aug, kpsois_aug)]
ia.imshow(ia.draw_grid(images_show))
gen_time = timeit.timeit(
"iaa.generate_jigsaw_destinations(10, 10, 2, rng)",
number=128,
setup=(
"import imgaug.augmenters as iaa; "
"import imgaug.random as iarandom; "
"rng = iarandom.RNG(0)"
)
)
print("Time to generate 128x dest:", gen_time)
destinations = iaa.generate_jigsaw_destinations(10, 10, 1, seed=1)
image_jig = iaa.apply_jigsaw(image, destinations)
ia.imshow(image_jig)
def main():
augs = [("iaa.Rot90(-1, keep_size=False)", iaa.Rot90(-1, keep_size=False)),
("iaa.Rot90(0, keep_size=False)", iaa.Rot90(0, keep_size=False)),
("iaa.Rot90(1, keep_size=False)", iaa.Rot90(1, keep_size=False)),
("iaa.Rot90(2, keep_size=False)", iaa.Rot90(2, keep_size=False)),
("iaa.Rot90(3, keep_size=False)", iaa.Rot90(3, keep_size=False)),
("iaa.Rot90(4, keep_size=False)", iaa.Rot90(4, keep_size=False)),
("iaa.Rot90(-1, keep_size=True)", iaa.Rot90(-1, keep_size=True)),
("iaa.Rot90(0, keep_size=True)", iaa.Rot90(0, keep_size=True)),
("iaa.Rot90(1, keep_size=True)", iaa.Rot90(1, keep_size=True)),
("iaa.Rot90(2, keep_size=True)", iaa.Rot90(2, keep_size=True)),
("iaa.Rot90(3, keep_size=True)", iaa.Rot90(3, keep_size=True)),
("iaa.Rot90(4, keep_size=True)", iaa.Rot90(4, keep_size=True)),
("iaa.Rot90([0, 1, 2, 3, 4], keep_size=False)",
iaa.Rot90([0, 1, 2, 3, 4], keep_size=False)),
("iaa.Rot90([0, 1, 2, 3, 4], keep_size=True)",
iaa.Rot90([0, 1, 2, 3, 4], keep_size=True)),
("iaa.Rot90((0, 4), keep_size=False)",
iaa.Rot90((0, 4), keep_size=False)),
("iaa.Rot90((0, 4), keep_size=True)",
iaa.Rot90((0, 4), keep_size=True)),
("iaa.Rot90((1, 3), keep_size=False)",
iaa.Rot90((1, 3), keep_size=False)),
("iaa.Rot90((1, 3), keep_size=True)",
iaa.Rot90((1, 3), keep_size=True))]
image = ia.data.quokka(0.25)
print("--------")
print("Image + Keypoints")
print("--------")
kps = ia.quokka_keypoints(0.25)
for name, aug in augs:
print(name, "...")
aug_det = aug.to_deterministic()
images_aug = aug_det.augment_images([image] * 16)
kps_aug = aug_det.augment_keypoints([kps] * 16)
images_aug = [
kps_aug_i.draw_on_image(image_aug_i, size=5)
for image_aug_i, kps_aug_i in zip(images_aug, kps_aug)
]
ia.imshow(ia.draw_grid(images_aug))
print("--------")
print("Image + Heatmaps (low res)")
print("--------")
hms = ia.quokka_heatmap(0.10)
for name, aug in augs:
print(name, "...")
aug_det = aug.to_deterministic()
images_aug = aug_det.augment_images([image] * 16)
hms_aug = aug_det.augment_heatmaps([hms] * 16)
images_aug = [
hms_aug_i.draw_on_image(image_aug_i)[0]
for image_aug_i, hms_aug_i in zip(images_aug, hms_aug)
]
ia.imshow(ia.draw_grid(images_aug))
def main():
augs = [
("iaa.Rot90(-1, keep_size=False)", iaa.Rot90(-1, keep_size=False)),
("iaa.Rot90(0, keep_size=False)", iaa.Rot90(0, keep_size=False)),
("iaa.Rot90(1, keep_size=False)", iaa.Rot90(1, keep_size=False)),
("iaa.Rot90(2, keep_size=False)", iaa.Rot90(2, keep_size=False)),
("iaa.Rot90(3, keep_size=False)", iaa.Rot90(3, keep_size=False)),
("iaa.Rot90(4, keep_size=False)", iaa.Rot90(4, keep_size=False)),
("iaa.Rot90(-1, keep_size=True)", iaa.Rot90(-1, keep_size=True)),
("iaa.Rot90(0, keep_size=True)", iaa.Rot90(0, keep_size=True)),
("iaa.Rot90(1, keep_size=True)", iaa.Rot90(1, keep_size=True)),
("iaa.Rot90(2, keep_size=True)", iaa.Rot90(2, keep_size=True)),
("iaa.Rot90(3, keep_size=True)", iaa.Rot90(3, keep_size=True)),
("iaa.Rot90(4, keep_size=True)", iaa.Rot90(4, keep_size=True)),
("iaa.Rot90([0, 1, 2, 3, 4], keep_size=False)", iaa.Rot90([0, 1, 2, 3, 4], keep_size=False)),
("iaa.Rot90([0, 1, 2, 3, 4], keep_size=True)", iaa.Rot90([0, 1, 2, 3, 4], keep_size=True)),
("iaa.Rot90((0, 4), keep_size=False)", iaa.Rot90((0, 4), keep_size=False)),
("iaa.Rot90((0, 4), keep_size=True)", iaa.Rot90((0, 4), keep_size=True)),
("iaa.Rot90((1, 3), keep_size=False)", iaa.Rot90((1, 3), keep_size=False)),
("iaa.Rot90((1, 3), keep_size=True)", iaa.Rot90((1, 3), keep_size=True))
]
image = ia.quokka(0.25)
print("--------")
print("Image + Keypoints")
print("--------")
kps = ia.quokka_keypoints(0.25)
for name, aug in augs:
print(name, "...")
aug_det = aug.to_deterministic()
images_aug = aug_det.augment_images([image] * 16)
kps_aug = aug_det.augment_keypoints([kps] * 16)
images_aug = [kps_aug_i.draw_on_image(image_aug_i, size=5)
for image_aug_i, kps_aug_i in zip(images_aug, kps_aug)]
ia.imshow(ia.draw_grid(images_aug))
print("--------")
print("Image + Heatmaps (low res)")
print("--------")
hms = ia.quokka_heatmap(0.10)
for name, aug in augs:
print(name, "...")
aug_det = aug.to_deterministic()
images_aug = aug_det.augment_images([image] * 16)
hms_aug = aug_det.augment_heatmaps([hms] * 16)
images_aug = [hms_aug_i.draw_on_image(image_aug_i)[0]
for image_aug_i, hms_aug_i in zip(images_aug, hms_aug)]
ia.imshow(ia.draw_grid(images_aug))