Removing sections of contours that overlap!

SpeedySuperPixels.py

@@ -19,19 +19,31 @@ class SuperContour(object):
     def __init__(self, width=32):
         self.width = width
 
-    def grid_contours(self, contours, heir):
+    def grid_contours(self, frame, contours, heir):
         result_cont, result_heir, result_rois = [], [], []
+
         # todo: mix grid with contours to form super pixels!
         # todo: remove any regions of overlap!
+
+        logger.debug('checking and removing overlap...')
+        msk_all = numpy.zeros(frame.shape[:2], dtype=frame.dtype)
+        for contour in contours:
+            msk = numpy.zeros(frame.shape, dtype=frame.dtype)
+            cv2.drawContours(msk, [contour], -1, 255, -1)
+            msk = cv2.bitwise_and(msk, cv2.bitwise_not(msk_all))
+            if msk.sum() != 0:
+                result_cont.append(msk)
+                msk_all = numpy.min(255, cv2.add(msk_all, msk))
+        logger.debug('grid contours complete')
         result_heir = heir
         return result_cont, result_heir, result_rois
 
     def process(self, frame):
         frame_gry = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
         contours, heir = cv2.findContours(frame_gry, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
-        contours, heir, rois = self.grid_contours(contours, heir)
+        contours, heir, rois = self.grid_contours(frame, contours, heir)
         for i in range(len(contours)):
             roi, contour = rois[i], contours[i]
-            mask = numpy.zeros(frame.shape, dtype=frame.dtype)
+            mask = numpy.zeros(frame.shape[:2], dtype=frame.dtype)
             cv2.drawContours(mask, [contour], -1, 255, -1)
             yield roi, contour
\ No newline at end of file