diff --git a/ContourExctractor.py b/ContourExctractor.py index 03ea153..410784a 100644 --- a/ContourExctractor.py +++ b/ContourExctractor.py @@ -83,6 +83,7 @@ class ContourExtractor: #cv2.imshow( "annotated", frame ) #cv2.waitKey(10) & 0XFF self.extractedContours = extractedContours + return extractedContours def displayContours(self): diff --git a/__pycache__/ContourExctractor.cpython-37.pyc b/__pycache__/ContourExctractor.cpython-37.pyc index 91f68bb..6be29d4 100644 Binary files a/__pycache__/ContourExctractor.cpython-37.pyc and b/__pycache__/ContourExctractor.cpython-37.pyc differ diff --git a/main.py b/main.py index 49f460c..2190b28 100644 --- a/main.py +++ b/main.py @@ -11,18 +11,11 @@ from LayerFactory import LayerFactory def demo(): print("startup") - footagePath = os.path.join(os.path.dirname(__file__), "./generate test footage/out.mp4") - start = time.time() - contourExtractor = ContourExtractor() - contourExtractor.extractContours(footagePath) + + footagePath = os.path.join(os.path.dirname(__file__), "./generate test footage/3.MP4") + contours = ContourExtractor().extractContours(footagePath) print("Time consumed in working: ",time.time() - start) - - #frames = contourExtractor.exportContours() - #Exporter().export(frames,os.path.join(os.path.dirname(__file__), "./short.mp4")) - - contours = contourExtractor.getextractedContours() - layerFactory = LayerFactory(contours) Exporter().exportOverlayed(layerFactory.layers, os.path.join(os.path.dirname(__file__), "./short.mp4")) diff --git a/short.mp4 b/short.mp4 index 30dfecb..faa60bd 100644 Binary files a/short.mp4 and b/short.mp4 differ