From 90ecff91f4c88817c0e05f7fda1d765d55e944e9 Mon Sep 17 00:00:00 2001 From: Askill Date: Mon, 15 Aug 2022 12:21:20 +0200 Subject: [PATCH] formatting --- Application/VideoReader.py | 5 ++--- main.py | 4 +++- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/Application/VideoReader.py b/Application/VideoReader.py index 8e251be..b7f670b 100644 --- a/Application/VideoReader.py +++ b/Application/VideoReader.py @@ -19,7 +19,7 @@ class VideoReader: # buffer data struct: # buffer = Queue([(frameNumber, frame), ]) self.buffer = multiprocessing.Queue(config["videoBufferLength"]) - #self.vc = cv2.VideoCapture(videoPath) + # self.vc = cv2.VideoCapture(videoPath) self.stopped = False self.getWH() self.calcFPS() @@ -90,7 +90,6 @@ class VideoReader: self.lastFrame = self.listOfFrames[0] self.buffer.put((self.lastFrame, None)) - def videoEnded(self): if self.stopped and self.buffer.empty(): return True @@ -130,5 +129,5 @@ class VideoReader: res, image = vc.read() self.w = image.shape[1] self.h = image.shape[0] - + return (self.w, self.h) diff --git a/main.py b/main.py index 5aa9fca..3fc6c87 100644 --- a/main.py +++ b/main.py @@ -30,7 +30,9 @@ def main(config): if len(layerManager.layers) == 0: exit(1) - heatmap = HeatMap(config["w"], config["h"], [contour for layer in layerManager.layers for contour in layer.bounds], 1920 / config["resizeWidth"]) + heatmap = HeatMap( + config["w"], config["h"], [contour for layer in layerManager.layers for contour in layer.bounds], 1920 / config["resizeWidth"] + ) heatmap.showImage() print(f"Exporting {len(contours)} Contours and {len(layerManager.layers)} Layers")