smol change

This commit is contained in:
Askill 2020-09-30 19:50:43 +02:00
parent 5837e6330e
commit de85715787
4 changed files with 4 additions and 10 deletions

View File

@ -83,6 +83,7 @@ class ContourExtractor:
#cv2.imshow( "annotated", frame ) #cv2.imshow( "annotated", frame )
#cv2.waitKey(10) & 0XFF #cv2.waitKey(10) & 0XFF
self.extractedContours = extractedContours self.extractedContours = extractedContours
return extractedContours
def displayContours(self): def displayContours(self):

13
main.py
View File

@ -11,18 +11,11 @@ from LayerFactory import LayerFactory
def demo(): def demo():
print("startup") print("startup")
footagePath = os.path.join(os.path.dirname(__file__), "./generate test footage/out.mp4")
start = time.time() 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) 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) layerFactory = LayerFactory(contours)
Exporter().exportOverlayed(layerFactory.layers, os.path.join(os.path.dirname(__file__), "./short.mp4")) Exporter().exportOverlayed(layerFactory.layers, os.path.join(os.path.dirname(__file__), "./short.mp4"))

BIN
short.mp4

Binary file not shown.