Beispiel #1
0
if __name__ == "__main__":
    # DEBUG Prueba de las funciones (No se usara, Archivo usado como libreria)
    titulo = "Detector YoLo"
    dims = (0, 0)
    # Config.Fullscreen(titulo)
    out = None
    # Para capturar la salida
    if Config.VidProp.guardar:
        from Config import VidProp
        out = cv.VideoWriter(f"Salida/{titulo}.avi", VidProp.fourcc,
                             VidProp.fps, VidProp.resolu)
        # print(out.get(2)) # TODO
    # Abre el video y almacena las dimesiones
    cap = cv.VideoCapture(Config.VidProp.source)
    dims = Utiles.dimensiones_video(cap)

    # Crea la red neural
    modelo = genera_DNN()

    # Extrae las capas de YoLo
    layer_names = modelo.getLayerNames()
    layer_names = [
        layer_names[i[0] - 1] for i in modelo.getUnconnectedOutLayers()
    ]
    labels = genera_labels()
    colores = genera_colores(labels)
    fps = FPS().start()

    while cap.isOpened():
        ret, image = cap.read()