3.6 KiB
Clasificar una imagen usando un clasificador de imágenes basado en IoT Edge - Hardware IoT Virtual y Raspberry Pi
En esta parte de la lección, usarás el clasificador de imágenes que se ejecuta en el dispositivo IoT Edge.
Usar el clasificador de IoT Edge
El dispositivo IoT puede ser redirigido para usar el clasificador de imágenes de IoT Edge. La URL para el clasificador de imágenes es http://<IP address or name>/image
, reemplazando <IP address or name>
con la dirección IP o el nombre del host del ordenador que ejecuta IoT Edge.
La biblioteca de Python para Custom Vision solo funciona con modelos alojados en la nube, no con modelos alojados en IoT Edge. Esto significa que necesitarás usar la API REST para llamar al clasificador.
Tarea - usar el clasificador de IoT Edge
-
Abre el proyecto
fruit-quality-detector
en VS Code si aún no está abierto. Si estás usando un dispositivo IoT virtual, asegúrate de que el entorno virtual esté activado. -
Abre el archivo
app.py
y elimina las declaraciones de importación deazure.cognitiveservices.vision.customvision.prediction
ymsrest.authentication
. -
Agrega la siguiente importación al inicio del archivo:
import requests
-
Elimina todo el código después de que la imagen se guarda en un archivo, desde
image_file.write(image.read())
hasta el final del archivo. -
Agrega el siguiente código al final del archivo:
prediction_url = '<URL>' headers = { 'Content-Type' : 'application/octet-stream' } image.seek(0) response = requests.post(prediction_url, headers=headers, data=image) results = response.json() for prediction in results['predictions']: print(f'{prediction["tagName"]}:\t{prediction["probability"] * 100:.2f}%')
Reemplaza
<URL>
con la URL de tu clasificador.Este código realiza una solicitud POST REST al clasificador, enviando la imagen como el cuerpo de la solicitud. Los resultados regresan como JSON, y este se decodifica para imprimir las probabilidades.
-
Ejecuta tu código, con tu cámara apuntando a alguna fruta, o un conjunto de imágenes apropiado, o fruta visible en tu cámara web si estás usando hardware IoT virtual. Verás la salida en la consola:
(.venv) ➜ fruit-quality-detector python app.py ripe: 56.84% unripe: 43.16%
💁 Puedes encontrar este código en la carpeta code-classify/pi o code-classify/virtual-iot-device.
😀 ¡Tu programa clasificador de calidad de frutas fue un éxito!
Descargo de responsabilidad:
Este documento ha sido traducido utilizando el servicio de traducción automática Co-op Translator. Aunque nos esforzamos por garantizar la precisión, tenga en cuenta que las traducciones automatizadas pueden contener errores o imprecisiones. El documento original en su idioma nativo debe considerarse como la fuente autorizada. Para información crítica, se recomienda una traducción profesional realizada por humanos. No nos hacemos responsables de malentendidos o interpretaciones erróneas que puedan surgir del uso de esta traducción.