Ejemplo n.º 1
0
    # Eliminar fondo de la imagen a color
    frame = FC.EliminarFondo(frame, profundidad, clipping_distance,
                             153)  # Eliminamos fondo

    # Iniciar a contar tiempo para contar FPS
    start_time = time.time()

    # Convertir a float32 para TF
    batch_data = tf.constant(FC.PreprocessToTF(frame))

    # Crear preicciones
    pred_bbox = infer(batch_data)

    # obtener configuracion de cuadros
    boxes, pred_conf = FC.ConfigurarCuadro(pred_bbox)

    boxes, scores, classes, valid_detections = tf.image.combined_non_max_suppression(
        boxes=tf.reshape(boxes, (tf.shape(boxes)[0], -1, 1, 4)),
        scores=tf.reshape(
            pred_conf, (tf.shape(pred_conf)[0], -1, tf.shape(pred_conf)[-1])),
        max_output_size_per_class=50,
        max_total_size=50,
    )
    pred_bbox = [
        boxes.numpy(),
        scores.numpy(),
        classes.numpy(),
        valid_detections.numpy()
    ]