Added draw contours utility function

colorbynumber/utils.py

@@ -1,4 +1,6 @@
+import cv2 as cv
 from matplotlib import pyplot as plt
+import numpy as np
 
 def show_image(image, cmap = None):
     if cmap:
@@ -8,3 +10,13 @@ def show_image(image, cmap = None):
     plt.axis('off')
     plt.show()
 
+def draw_contours(contour, image_width, image_height):
+    contours_image = np.ones((image_width, image_height), dtype = np.uint8)
+    if type(contour) == tuple:
+        # If the contour is a tuple, it is a tuple of multiple contours
+        cv.drawContours(contours_image, contour, -1, (0,255,0), 4)
+    else:
+        # Else, it is a single contour
+        cv.drawContours(contours_image, [contour], 0, (0,255,0), 4)
+    show_image(contours_image, cmap = 'gray')
+