marzo 9, 2022

VARIOS – Prueba de detección con TensorFlow

Written by

NUEVA PRUEBA CON TARJETA GRÁFICA NVIDIA

Una vez que se ha instalado TensorFlow en el ordenador vamos a probarlo con algún ejemplo. El tutorial aparece en el enlace

https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/auto_examples/index.html#examples

Aunque hay que hacer algunas modificaciones para conseguir que funcione.

1 – Descargar el código de ejemplo del anterior enlace. Vamos a probar el primero de ellos, aunque el procedimiento es igual para todos. El enlace está en la parte inferior de la página del enlace. Colocar el fichero en alguna carpeta del ordenador, por ejemplo C:\Proyectos

2 – Abrir una ventana de comandos, cambiar a la carpeta del ejemplo y activar el entorno de TensorFlow, introduciendo los siguientes comandos:

cd \Proyectos
conda activate tensorflow

3 – La versión del ejemplo usa una versión más antigua de la librería pyparsing que la instalada en Anaconda. Para cambiarla, introducir el siguiente comando:

conda install pyparsing==2.4.2

4 – El código del ejemplo está preparado para ser usado desde un entorno Jupiter, así que hay que editarlo para poder ver el resultado. Hay que añadir las siguientes líneas:

import matplotlib
matplotlib.use('TkAgg')

justo en el párrafo donde se usa la librería matplotlib.

Cambios en el código de ejemplo

5 – Con esto ya se puede compilar y comprobar el resultado, con el comando:

python plot_object_detection_saved_model_tf1.py

La primera vez tarda varios minutos porque tiene que descargar los modelos y el sistema de detección. Las siguientes veces es más rápido. Finalmente podemos ver el resultado con las dos imágenes etiquetadas:

Resultado del reconocimiento de imágenes
Detección de objetos en las imágenes de ejemplo

Para probar que el sistema funciona, tomamos otra imagen de Internet y la aplicamos al ejemplo. Para que funcione, añadimos la imagen en el código:

Cambio de imagen en el código

Volviendo a ejecutar el código con la instrucción de antes, éste es el resultado:

Detección de objetos en otra imagen

Prueba de detección con TensorFlow con tarjeta Nvidia

Hacemos la prueba también con el sistema con tarjeta gráfica Nvidia. Seguimos el mismo tutorial del enlace:

https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/auto_examples/index.html#examples

y hay que hacer también alguna modificación.

1 – Descargar el código de ejemplo del anterior enlace. Probamos el primero de ellos. El enlace está en la parte inferior de la página del enlace tras pulsar en el ejemplo seleccionado. Colocamos el fichero en la carpeta del ordenador C:\Proyectos\TensorFlow.

2 – Hay que editar el código para poder ver el resultado. Hay que añadir las siguientes líneas:

import matplotlib
matplotlib.use('TkAgg')

justo en el párrafo donde se usa la librería matplotlib.

Cambios en el código de ejemplo

3 – Se compila y comprueba el resultado, con el comando:

python plot_object_detection_saved_model_tf1.py

El resultado en la consola se produce más rápidamente:

Ahora el resultado se produce en 7,6 segundos. Es una ligera mejora respecto al caso anterior.

Resultado del reconocimiento de objetos

Category : VARIOS

Deja una respuesta

Tu dirección de correo electrónico no será publicada.

Proudly powered by WordPress and Sweet Tech Theme