Video-Summary/main.py

import os
import time

from Application.Classifiers import *
from Application.Config import Config
from Application.ContourExctractor import ContourExtractor
from Application.Exporter import Exporter
from Application.HeatMap import HeatMap
from Application.Importer import Importer
from Application.LayerFactory import LayerFactory
from Application.LayerManager import LayerManager
from Application.VideoReader import VideoReader


def main():
    startTotal = time.time()
    config = Config()

    fileName = "./x23-1.mp4"
    outputPath = os.path.join(os.path.dirname(__file__), "output")
    dirName = os.path.join(os.path.dirname(__file__), "generate test footage")

    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:
        layers, contours, masks = Importer(config).importRawData()
        layerFactory = LayerFactory(config)
        layers = layerFactory.extractLayers(contours, masks)

    layerManager = LayerManager(config, layers)
    layerManager.cleanLayers()

    # layerManager.tagLayers()
    layers = layerManager.layers
    if len(layers) == 0:
        exit(1)

    heatmap = HeatMap(config["w"], config["h"], [contour for layer in layers for contour in layer.bounds], 1920 / config["resizeWidth"])
    heatmap.showImage()

    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)


if __name__ == "__main__":
    main()