--- a
+++ b/mediaug/visulize.py
@@ -0,0 +1,41 @@
+from IPython.display import display
+from PIL import Image
+from matplotlib import pyplot as plt
+
+
+def show_image(img):
+    """ Display image in jupyter noteobok
+    Args:
+      img (np.array): The image
+    """
+    display(Image.fromarray(img))
+
+
+def plot_datapoint(dp):
+    """ plot a datapoint image and mask side by side
+    Args:
+      dp (Datapoint): datapoint
+    """
+    plt.figure(figsize=(15,15))
+    plt.subplot(121)
+    plt.imshow(dp.img)
+    plt.axis('off')
+    plt.subplot(122)
+    plt.imshow(dp.mask)
+    plt.axis('off')
+    plt.show()
+
+def plot_img_and_mask(img, mask):
+    """ plot image next to mask
+    Args:
+      img (np.array): image
+      mask (np.array): mask
+    """
+    plt.figure(figsize=(15,15))
+    plt.subplot(121)
+    plt.imshow(img)
+    plt.axis('off')
+    plt.subplot(122)
+    plt.imshow(mask)
+    plt.axis('off')
+    plt.show()
\ No newline at end of file