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()