labels = genera_labels()
colores = genera_colores(labels)
fps = FPS().start()
while cap.isOpened():
ret, image = cap.read()
if not ret:
print('Video file finished.')
break
umbral_confianza = 0.5
ROIs, predicciones, tipos, idxs = genera_predicciones(
modelo, layer_names, labels, image, umbral_confianza)
image = cv.UMat(image)
dibuja_ROIs(ROIs, predicciones, tipos, idxs)
if Config.VidProp.show_fps:
Utiles.dibuja_FPS(image, fps)
cv.imshow(titulo, image)
if (cv.waitKey(1) & 0xFF == 27):
break
if Config.VidProp.guardar:
Utiles.guardar(out, image)
cap.release()
if Config.VidProp.guardar:
out.release()
Utiles.mascara_destruidos(image, objetivos_destruidos)
ROIs = deteccion(image)
if (len(ROIs) > 1):
# Asigna solo los rectangulos contendores como objetivos
objetivos = ROIs[0]
# image = Utiles.atencion_blur(image, objetivos[0])
# image = Utiles.multi_atencion_blur(image, objetivos)
objetivo = Selector.objetivo_prioritario(p_actual, objetivos)
if (objetivo):
modo = Config.Modo.search_destroy
Utiles.dibuja_rectangulo(image, objetivo, Config.UI.cyan2)
image = Utiles.atencion_blur(image, objetivo)
else:
image = search_destroy(image)
cond = False
if (cond):
modo = Config.Modo.deteccion
if Config.VidProp.show_fps: Utiles.dibuja_FPS(image, fps)
if Config.VidProp.guardar: Utiles.guardar(out, image)
cv.imshow(titulo, image)
# TODO solo 10 frames pausa
# cv.waitKey(0)
# if count > 100: break
# count += 1
# TODO solo 10 frames
if (cv.waitKey(1) & 0xFF == 27): break
# cv.waitKey(0)
if Config.VidProp.guardar: out.release()
