single-shot.py

# This script is used to look for objects under a specific condition (at least 5 persons etc)
# The script reads a video from a message queue, classifies the objects in the video, and does a condition check.
# If condition is met, the video is being forwarded to a remote vault.
from exports.export_factory import ExportFactory
from integrations.integration_factory import IntegrationFactory
from projects.project_factory import ProjectFactory
from services.harvest_service import HarvestService

from utils.VariableClass import VariableClass
from utils.time_verbose_object import TimeVerbose

# Initialize the VariableClass object, which contains all the necessary environment variables.
var = VariableClass()


def init():
    # Service and Project initializations
    project = ProjectFactory().init()
    integration = IntegrationFactory().init()
    export = ExportFactory().init()
    harvest_service = HarvestService()

    # register to service
    harvest_service.register('project', project)
    harvest_service.register('integration', integration)
    harvest_service.register('export', export)

    # Open video-capture/recording using the video-path. Throw FileNotFoundError if cap is unable to open.
    time_verbose = TimeVerbose()
    video = harvest_service.open_video()

    if var.LOGGING:
        print(f'5. Classifying frames')
    if var.TIME_VERBOSE:
        time_verbose.add_preprocessing_time()

    # Evaluate the video
    save_dir = harvest_service.evaluate(video)

    # Upload dataset if True
    if var.DATASET_UPLOAD:
        integration.upload_dataset(save_dir)

    if var.TIME_VERBOSE:
        time_verbose.add_preprocessing_time()

    # Depending on the TIME_VERBOSE parameter, the time it took to classify the objects is printed.
    if var.TIME_VERBOSE:
        time_verbose.show_result()

    if var.LOGGING:
        print('8) Releasing video writer and closing video capture')
        print("\n\n")

    # video_out.release() if var.SAVE_VIDEO else None
    # cap.release()
    # if var.PLOT:
    #     cv2.destroyAllWindows()


# Run the init function.
init()