check_requirements.py

required_libs = {
    "gstreamer": "1.14.2".split("."),
    "tensorflow": "1.10.0".split("."),
    "opencv": "3.4.0".split(".")
}


def check_opencv():
    major, minor, _ = 0, 0, 0

    try:
        import cv2
        major, minor, _ = cv2.__version__.split(".")
    except Exception as e:
        raise ImportError("Can't import OpenCV")
        
    rmajor, rminor, _ = required_libs["opencv"]
    assert str(major) == str(rmajor), "Invalid OpenCV Version.Installed: {}.{}, Required: v{}.{}".format(major, minor, rmajor, rminor)


def check_tesorflow():
    major, minor, _ = 0, 0, 0

    try:
        import tensorflow as tf
        major, minor, _ = tf.__version__.split(".")
    except Exception as e:
        raise ImportError("Can't import Tensorflow")
        
    rmajor, rminor, _ = required_libs["tensorflow"]
    assert str(major) == str(rmajor) and str(minor) == str(rminor), \
        "Invalid Tensorflow Version. Installed: {}.{}. Required: v{}.{}".format(major, minor, rmajor, rminor)


def check_gstreamer():
    major, minor, micro = 0, 0, 0

    try:
        import gi
        gi.require_version('Gst', '1.0')
        gi.require_version('GstBase', '1.0')
        from gi.repository import Gst, GObject
        Gst.init(None)
        major, minor, micro, _ = Gst.version()
    except Exception as e:
        raise ImportError("Can't import Gstreamer")
        
    rmajor, rminor, rmicro = required_libs["gstreamer"]
    assert str(major) == str(rmajor) and str(minor) == str(rminor), \
        "Invalid Gstreamer Version. Installed: {}.{}.{}.Required: v{}.{}.{}".format(major, minor, micro, rmajor, rminor, rmicro)
    
check_gstreamer()
check_opencv()
check_tesorflow()