Link to this sectionUna guía sobre la exportación de modelos YOLO26 a TFLite para su despliegue#
Desplegar modelos de computer vision en dispositivos de borde o integrados requiere un formato que pueda garantizar un rendimiento fluido.
El formato de exportación TensorFlow Lite o TFLite te permite optimizar tus modelos Ultralytics YOLO26 para tareas como object detection y image classification en aplicaciones basadas en dispositivos de borde. En esta guía, recorreremos los pasos para convertir tus modelos al formato TFLite, facilitando que tus modelos funcionen bien en diversos dispositivos de borde.
Link to this section¿Por qué deberías exportar a TFLite?#
Introducido por Google en mayo de 2017 como parte de su marco de trabajo TensorFlow, TensorFlow Lite, o TFLite para abreviar, es un marco de aprendizaje profundo de código abierto diseñado para la inferencia en el dispositivo, también conocido como edge computing. Ofrece a los desarrolladores las herramientas necesarias para ejecutar sus modelos entrenados en dispositivos móviles, integrados e IoT, así como en computadoras tradicionales.
TensorFlow Lite es compatible con una amplia gama de plataformas, incluyendo Linux integrado, Android, iOS y microcontroladores (MCUs). Exportar tu modelo a TFLite hace que tus aplicaciones sean más rápidas, fiables y capaces de ejecutarse sin conexión.
Link to this sectionCaracterísticas clave de los modelos TFLite#
Los modelos TFLite ofrecen una amplia gama de características clave que permiten el aprendizaje automático en el dispositivo, ayudando a los desarrolladores a ejecutar sus modelos en dispositivos móviles, integrados y de borde:
-
Optimización en el dispositivo: TFLite se optimiza para ML en el dispositivo, reduciendo la latencia al procesar datos localmente, mejorando la privacidad al no transmitir datos personales y minimizando el tamaño del modelo para ahorrar espacio.
-
Soporte para múltiples plataformas: TFLite ofrece una amplia compatibilidad de plataformas, soportando Android, iOS, Linux integrado y microcontroladores.
-
Soporte para diversos lenguajes: TFLite es compatible con varios lenguajes de programación, incluyendo Java, Swift, Objective-C, C++ y Python.
-
Alto rendimiento: Logra un rendimiento superior mediante la aceleración de hardware y la optimización del modelo.
Link to this sectionOpciones de despliegue en TFLite#
Antes de revisar el código para exportar modelos YOLO26 al formato TFLite, comprendamos cómo se utilizan normalmente los modelos TFLite.
TFLite ofrece varias opciones de despliegue en el dispositivo para modelos de aprendizaje automático, incluyendo:
- Despliegue con Android e iOS: Tanto las aplicaciones de Android como las de iOS con TFLite pueden analizar transmisiones de cámaras y sensores de borde para detectar e identificar objetos. TFLite también ofrece bibliotecas nativas para iOS escritas en Swift y Objective-C. El diagrama de arquitectura a continuación muestra el proceso de desplegar un modelo entrenado en plataformas Android e iOS usando TensorFlow Lite.
-
Implementación con Linux integrado: Si la ejecución de inferencias en una Raspberry Pi usando la Guía de Ultralytics no cumple con los requisitos de velocidad para tu caso de uso, puedes usar un modelo TFLite exportado para acelerar los tiempos de inferencia. Además, es posible mejorar aún más el rendimiento utilizando un dispositivo Coral Edge TPU.
-
Despliegue con microcontroladores: Los modelos TFLite también pueden desplegarse en microcontroladores y otros dispositivos con solo unos pocos kilobytes de memoria. El núcleo de ejecución cabe en solo 16 KB en un Arm Cortex M3 y puede ejecutar muchos modelos básicos. No requiere soporte de sistema operativo, ninguna biblioteca estándar de C o C++ ni asignación de memoria dinámica.
Link to this sectionExportar a TFLite: Convirtiendo tu modelo YOLO26#
Puedes mejorar la eficiencia de ejecución del modelo en el dispositivo y optimizar el rendimiento convirtiendo tus modelos al formato TFLite.
Link to this sectionInstalación#
Para instalar los paquetes necesarios, ejecuta:
# Install the required package for YOLO26
pip install ultralyticsPara instrucciones detalladas y mejores prácticas relacionadas con el proceso de instalación, consulta nuestra Guía de instalación de Ultralytics. Mientras instalas los paquetes requeridos para YOLO26, si encuentras alguna dificultad, consulta nuestra Guía de problemas comunes para obtener soluciones y consejos.
Link to this sectionUso#
Todos los modelos de Ultralytics YOLO26 están diseñados para admitir la exportación desde el primer momento, facilitando su integración en tu flujo de trabajo de despliegue preferido. Puedes ver la lista completa de formatos de exportación admitidos y las opciones de configuración para elegir la mejor configuración para tu aplicación.
El formato TFLite soporta los modos Export, Predict y Validate. Exporta tu modelo, luego carga el modelo exportado para ejecutar inferencias o validar su precisión.
from ultralytics import YOLO
# Load a YOLO26 model
model = YOLO("yolo26n.pt")
# Export the model to TFLite format
model.export(format="tflite") # creates 'yolo26n_float32.tflite'from ultralytics import YOLO
# Load the exported TFLite model
model = YOLO("yolo26n_float32.tflite")
# Run inference
results = model("https://ultralytics.com/images/bus.jpg")from ultralytics import YOLO
# Load the exported TFLite model
model = YOLO("yolo26n_float32.tflite")
# Validate accuracy on the COCO8 dataset
metrics = model.val(data="coco8.yaml")Link to this sectionArgumentos de exportación#
| Argumento | Tipo | Predeterminado | Descripción |
|---|---|---|---|
format | str | 'tflite' | Formato de destino para el modelo exportado, definiendo la compatibilidad con diversos entornos de despliegue. |
imgsz | int o tuple | 640 | Tamaño de imagen deseado para la entrada del modelo. Puede ser un número entero para imágenes cuadradas o una tupla (height, width) para dimensiones específicas. |
half | bool | False | Habilita la cuantización FP16 (media precisión), reduciendo el tamaño del modelo y potencialmente acelerando la inferencia en hardware compatible. |
int8 | bool | False | Activa la cuantización INT8, comprimiendo aún más el modelo y acelerando la inferencia con una mínima pérdida de precisión, principalmente para dispositivos de borde. |
nms | bool | False | Añade la supresión de no máximos (NMS), esencial para el posprocesamiento preciso y eficiente de detecciones. |
batch | int | 1 | Especifica el tamaño de inferencia por lote del modelo exportado o el número máximo de imágenes que el modelo procesará simultáneamente en el modo predict. |
data | str | 'coco8.yaml' | Ruta al archivo de configuración del dataset (predeterminado: coco8.yaml), esencial para la cuantización. |
fraction | float | 1.0 | Especifica la fracción del conjunto de datos a utilizar para la calibración de cuantización INT8. Permite calibrar en un subconjunto del conjunto de datos completo, útil para experimentos o cuando los recursos son limitados. Si no se especifica con INT8 habilitado, se utilizará el conjunto de datos completo. |
device | str | None | Especifica el dispositivo para la exportación: CPU (device=cpu), MPS para Apple silicon (device=mps). |
Para obtener más detalles sobre el proceso de exportación, visita la página de documentación de Ultralytics sobre la exportación.
Link to this sectionDesplegando modelos YOLO26 TFLite exportados#
Después de exportar con éxito tus modelos Ultralytics YOLO26 al formato TFLite, ya puedes desplegarlos. El primer paso recomendado para ejecutar un modelo TFLite es utilizar el método YOLO("model.tflite"), como se describe en el fragmento de código de uso anterior. Sin embargo, para obtener instrucciones detalladas sobre cómo desplegar tus modelos TFLite en otros entornos, echa un vistazo a los siguientes recursos:
-
Android: Una guía de inicio rápido para integrar TensorFlow Lite en aplicaciones Android, que proporciona pasos fáciles de seguir para configurar y ejecutar modelos de machine learning.
-
iOS: Consulta esta guía detallada para desarrolladores sobre cómo integrar y desplegar modelos TensorFlow Lite en aplicaciones iOS, ofreciendo instrucciones paso a paso y recursos.
-
Ejemplos integrales: Esta página ofrece una visión general de varios ejemplos de TensorFlow Lite, mostrando aplicaciones prácticas y tutoriales diseñados para ayudar a los desarrolladores a implementar TensorFlow Lite en sus proyectos de aprendizaje automático en dispositivos móviles y de borde.
Link to this sectionResumen#
En esta guía, nos hemos centrado en cómo exportar al formato TFLite. Al convertir tus modelos Ultralytics YOLO26 al formato de modelo TFLite, puedes mejorar la eficiencia y la velocidad de los modelos YOLO26, haciéndolos más efectivos y adecuados para entornos de edge computing.
Para más detalles sobre su uso, visita la documentación oficial de TFLite.
Además, si tienes curiosidad sobre otras integraciones de Ultralytics YOLO26, echa un vistazo a nuestra página de guía de integración. Encontrarás mucha información útil y perspectivas allí.
Link to this sectionFAQ#
Link to this section¿Cómo exporto un modelo YOLO26 al formato TFLite?#
Para exportar un modelo YOLO26 al formato TFLite, puedes usar la biblioteca de Ultralytics. Primero, instala el paquete requerido usando:
pip install ultralyticsLuego, usa el siguiente fragmento de código para exportar tu modelo:
from ultralytics import YOLO
# Load a YOLO26 model
model = YOLO("yolo26n.pt")
# Export the model to TFLite format
model.export(format="tflite") # creates 'yolo26n_float32.tflite'Para usuarios de CLI, puedes lograr esto con:
yolo export model=yolo26n.pt format=tflite # creates 'yolo26n_float32.tflite'Para más detalles, visita la guía de exportación de Ultralytics.
Link to this section¿Cuáles son los beneficios de usar TensorFlow Lite para el despliegue de modelos YOLO26?#
TensorFlow Lite (TFLite) es un marco de deep learning de código abierto diseñado para la inferencia en el dispositivo, lo que lo hace ideal para desplegar modelos YOLO26 en dispositivos móviles, integrados e IoT. Los beneficios clave incluyen:
- Optimización en el dispositivo: Minimiza la latencia y mejora la privacidad procesando los datos localmente.
- Compatibilidad de plataforma: Soporta Android, iOS, Linux integrado y MCU.
- Rendimiento: Utiliza aceleración de hardware para optimizar la velocidad y eficiencia del modelo.
Para aprender más, consulta la guía de TFLite.
Link to this section¿Es posible ejecutar modelos YOLO26 TFLite en Raspberry Pi?#
Sí, puedes ejecutar modelos YOLO26 TFLite en Raspberry Pi para mejorar las velocidades de inferencia. Primero, exporta tu modelo al formato TFLite como se explicó anteriormente. Luego, usa una herramienta como TensorFlow Lite Interpreter para ejecutar el modelo en tu Raspberry Pi.
Para optimizaciones adicionales, podrías considerar usar Coral Edge TPU. Para pasos detallados, consulta nuestra guía de despliegue en Raspberry Pi y la guía de integración de Edge TPU.
Link to this section¿Puedo usar modelos TFLite en microcontroladores para predicciones con YOLO26?#
Sí, TFLite soporta el despliegue en microcontroladores con recursos limitados. El núcleo de ejecución de TFLite requiere solo 16 KB de memoria en un Arm Cortex M3 y puede ejecutar modelos YOLO26 básicos. Esto lo hace adecuado para su despliegue en dispositivos con potencia computacional y memoria mínimas.
Para empezar, visita la guía de TFLite Micro para microcontroladores.
Link to this section¿Qué plataformas son compatibles con los modelos YOLO26 exportados a TFLite?#
TensorFlow Lite proporciona una amplia compatibilidad de plataformas, permitiéndote desplegar modelos YOLO26 en una amplia gama de dispositivos, incluyendo:
- Android e iOS: Soporte nativo a través de las bibliotecas de TFLite para Android e iOS.
- Linux integrado: Ideal para computadoras de placa única como Raspberry Pi.
- Microcontroladores: Adecuado para MCUs con recursos limitados.
Para más información sobre las opciones de despliegue, consulta nuestra guía de despliegue detallada.
Link to this section¿Cómo soluciono problemas comunes durante la exportación de modelos YOLO26 a TFLite?#
Si encuentras errores al exportar modelos YOLO26 a TFLite, las soluciones comunes incluyen:
- Comprobar la compatibilidad del paquete: Asegúrate de estar usando versiones compatibles de Ultralytics y TensorFlow. Consulta nuestra guía de instalación.
- Soporte del modelo: Verifica que el modelo YOLO26 específico soporte la exportación a TFLite comprobando la página de documentación de exportación de Ultralytics.
- Problemas de cuantificación: Cuando uses cuantificación INT8, asegúrate de que la ruta de tu conjunto de datos esté correctamente especificada en el parámetro
data.
Para consejos adicionales de solución de problemas, visita nuestra Guía de problemas comunes.