Video-Summary/main.py

40 lines
1.2 KiB
Python
Raw Normal View History

2020-09-22 18:25:06 +00:00
import os
import time
from ContourExctractor import ContourExtractor
from Exporter import Exporter
2020-09-24 20:48:04 +00:00
from LayerFactory import LayerFactory
2020-10-05 07:43:27 +00:00
from Analyzer import Analyzer
2020-10-08 20:26:29 +00:00
from VideoReader import VideoReader
2020-10-03 22:27:36 +00:00
import cv2
2020-09-22 18:25:06 +00:00
#TODO
# finden von relevanten Stellen anhand von zu findenen metriken für vergleichsbilder
2020-09-20 20:01:54 +00:00
def demo():
2020-09-20 20:01:54 +00:00
print("startup")
resizeWidth = 256
maxLayerLength = 20*30
2020-10-08 20:26:29 +00:00
minLayerLength = 30
2020-09-22 18:25:06 +00:00
start = time.time()
2020-09-24 13:47:49 +00:00
2020-10-05 20:24:38 +00:00
footagePath = os.path.join(os.path.dirname(__file__), "./generate test footage/3.mp4")
2020-10-07 12:32:41 +00:00
#analyzer = Analyzer(footagePath)
#print("Time consumed reading video: ", time.time() - start)
2020-10-08 20:26:29 +00:00
2020-10-03 22:27:36 +00:00
contours = ContourExtractor().extractContours(footagePath, resizeWidth)
2020-10-11 12:13:27 +00:00
print("Time consumed extracting: ", time.time() - start)
2020-09-28 20:28:23 +00:00
layerFactory = LayerFactory(contours)
2020-10-07 12:32:41 +00:00
layerFactory.freeData(maxLayerLength, minLayerLength)
2020-10-05 20:24:38 +00:00
print("sort Layers")
2020-10-03 22:27:36 +00:00
layerFactory.sortLayers()
Exporter().exportOverlayed(layerFactory.layers,footagePath, os.path.join(os.path.dirname(__file__), "./short.mp4"), resizeWidth)
2020-10-03 22:27:36 +00:00
print("Total time: ", time.time() - start)
2020-10-05 20:24:38 +00:00
def init():
print("not needed yet")
2020-09-20 20:01:54 +00:00
if __name__ == "__main__":
demo()
2020-09-22 18:25:06 +00:00