Visualize augmented image and segmentation mask with imgaug

img_aug_visualize_mask.py

# Import matplotlib for visualization
from matplotlib import pyplot as plt

# Visualize original image, augmented image, and segmentation map
side_by_side = np.hstack([
    segmap.draw_on_image(image), # show blend of original image and segmentation map
    segmap_aug.draw_on_image(image_aug),  # show blend of augmented image and segmentation map
    segmap_aug.draw()  # show only the augmented segmentation map
])

fig, ax = plt.subplots(figsize=(10, 7))
ax.axis('off')
plt.title('Augmentations for segmentation masks')
ax.imshow(side_by_side)