From ffaf2813852923371403a5f780efe44ce752099d Mon Sep 17 00:00:00 2001 From: Askill Date: Sat, 26 Dec 2020 12:44:53 +0100 Subject: [PATCH] neetness --- Application/ContourExctractor.py | 1 - main.py | 6 ------ 2 files changed, 7 deletions(-) diff --git a/Application/ContourExctractor.py b/Application/ContourExctractor.py index 6ffa1ea..914454d 100644 --- a/Application/ContourExctractor.py +++ b/Application/ContourExctractor.py @@ -30,7 +30,6 @@ class ContourExtractor: def getExtractedMasks(self): return self.extractedMasks - def __init__(self, config): self.frameBuffer = Queue(16) self.extractedContours = dict() diff --git a/main.py b/main.py index f5322df..4305812 100644 --- a/main.py +++ b/main.py @@ -12,7 +12,6 @@ from Application.Classifiers import * def main(): startTotal = time.time() - start = startTotal config = Config() fileName = "x23.mp4" @@ -21,14 +20,11 @@ def main(): config["inputPath"] = os.path.join(dirName, fileName) config["outputPath"] = os.path.join(outputPath, fileName) - config["importPath"] = os.path.join(outputPath, fileName.split(".")[0] + ".txt") - config["w"], config["h"] = VideoReader(config).getWH() if not os.path.exists(config["importPath"]): contours, masks = ContourExtractor(config).extractContours() - layerFactory = LayerFactory(config) layers = layerFactory.extractLayers(contours, masks) else: @@ -48,9 +44,7 @@ def main(): exporter = Exporter(config) print(f"Exporting {len(contours)} Contours and {len(layers)} Layers") exporter.export(layers, contours, masks, raw=True, overlayed=True) - print("Total time: ", time.time() - startTotal) - exit(0) if __name__ == "__main__": main()