Como exportar para TF GraphDef a partir do YOLO26 para implementação
Ao implementar modelos de visão computacional de ponta, como o YOLO26, em diferentes ambientes, você pode encontrar problemas de compatibilidade. O TensorFlow GraphDef, ou TF GraphDef, da Google oferece uma solução ao fornecer uma representação serializada e independente de plataforma do seu modelo. Ao usar o formato de modelo TF GraphDef, podes implementar o teu modelo YOLO26 em ambientes onde o ecossistema TensorFlow completo pode não estar disponível, como em dispositivos móveis ou hardware especializado.
Neste guia, vamos conduzir-te passo a passo sobre como exportar os teus modelos Ultralytics YOLO26 para o formato de modelo TF GraphDef. Ao converter o teu modelo, podes simplificar a implementação e utilizar as capacidades de visão computacional do YOLO26 numa gama mais ampla de aplicações e plataformas.
Por que deves exportar para TF GraphDef?
O TF GraphDef é um componente poderoso do ecossistema TensorFlow desenvolvido pela Google. Pode ser usado para otimizar e implementar modelos como o YOLO26. Exportar para TF GraphDef permite-te mover modelos da pesquisa para aplicações do mundo real. Permite que os modelos corram em ambientes sem a framework TensorFlow completa.
O formato GraphDef representa o modelo como um grafo de computação serializado. Isto possibilita várias técnicas de otimização, como 'constant folding', quantização e transformações de grafo. Estas otimizações garantem uma execução eficiente, menor uso de memória e velocidades de inferência mais rápidas.
Os modelos GraphDef podem utilizar aceleradores de hardware como GPUs, TPUs e chips de IA, desbloqueando ganhos de desempenho significativos para o pipeline de inferência do YOLO26. O formato TF GraphDef cria um pacote autónomo com o modelo e as suas dependências, simplificando a implementação e a integração em diversos sistemas.
Principais características dos modelos TF GraphDef
O TF GraphDef oferece características distintas para otimizar a implementação de modelos e a otimização.
Aqui tens uma visão das suas principais características:
-
Serialização de Modelo: O TF GraphDef fornece uma forma de serializar e armazenar modelos TensorFlow num formato independente de plataforma. Esta representação serializada permite-te carregar e executar os teus modelos sem a base de código Python original, tornando a implementação mais fácil.
-
Otimização de Grafo: O TF GraphDef permite a otimização de grafos computacionais. Estas otimizações podem aumentar o desempenho ao simplificar o fluxo de execução, reduzir redundâncias e adaptar operações para se adequarem a hardware específico.
-
Flexibilidade de Implementação: Os modelos exportados para o formato GraphDef podem ser usados em vários ambientes, incluindo dispositivos com recursos limitados, navegadores web e sistemas com hardware especializado. Isto abre possibilidades para uma implementação mais ampla dos teus modelos TensorFlow.
-
Foco em Produção: O GraphDef é concebido para implementação em produção. Suporta execução eficiente, funcionalidades de serialização e otimizações que se alinham com casos de uso do mundo real.
Opções de Implementação com TF GraphDef
Antes de mergulharmos no processo de exportar modelos YOLO26 para TF GraphDef, vamos dar uma vista de olhos em algumas situações típicas de implementação onde este formato é usado.
Aqui está como podes implementar com TF GraphDef de forma eficiente através de várias plataformas.
-
TensorFlow Serving: Esta framework foi concebida para implementar modelos TensorFlow em ambientes de produção. O TensorFlow Serving oferece gestão de modelos, controlo de versões e a infraestrutura para servir modelos de forma eficiente em larga escala. É uma forma simples de integrar os teus modelos baseados em GraphDef em serviços web ou APIs de produção.
-
Dispositivos Móveis e Integrados: Com ferramentas como o TensorFlow Lite, podes converter modelos TF GraphDef para formatos otimizados para smartphones, tablets e vários dispositivos integrados. Os teus modelos podem então ser usados para inferência no próprio dispositivo, onde a execução é feita localmente, muitas vezes proporcionando ganhos de desempenho e capacidades offline.
-
Navegadores Web: O TensorFlow.js permite a implementação de modelos TF GraphDef diretamente nos navegadores web. Abre o caminho para aplicações de deteção de objetos em tempo real a correr no lado do cliente, usando as capacidades do YOLO26 através de JavaScript.
-
Hardware Especializado: A natureza independente de plataforma do TF GraphDef permite que este direcione hardware personalizado, como aceleradores e TPUs (Tensor Processing Units). Estes dispositivos podem fornecer vantagens de desempenho para modelos computacionalmente intensivos.
Exportar Modelos YOLO26 para TF GraphDef
Podes converter o teu modelo de deteção de objetos YOLO26 para o formato TF GraphDef, que é compatível com vários sistemas, para melhorar o seu desempenho em várias plataformas.
Instalação
Para instalar o pacote necessário, execute:
# Install the required package for YOLO26
pip install ultralyticsPara instruções detalhadas e práticas recomendadas relacionadas ao processo de instalação, consulte nosso guia de instalação da Ultralytics. Ao instalar os pacotes necessários para o YOLO26, se você encontrar alguma dificuldade, consulte nosso guia de problemas comuns para obter soluções e dicas.
Utilização
Todos os modelos Ultralytics YOLO26 foram concebidos para suportar a exportação imediata, facilitando a sua integração no teu fluxo de trabalho de implementação preferido. Podes ver a lista completa de formatos de exportação suportados e opções de configuração para escolheres a melhor configuração para a tua aplicação.
from ultralytics import YOLO
# Load the YOLO26 model
model = YOLO("yolo26n.pt")
# Export the model to TF GraphDef format
model.export(format="pb") # creates 'yolo26n.pb'
# Load the exported TF GraphDef model
tf_graphdef_model = YOLO("yolo26n.pb")
# Run inference
results = tf_graphdef_model("https://ultralytics.com/images/bus.jpg")Argumentos de Exportação
| Argumento | Tipo | Predefinição | Descrição |
|---|---|---|---|
format | str | 'pb' | Formato de destino para o modelo exportado, definindo a compatibilidade com vários ambientes de implantação. |
imgsz | int ou tuple | 640 | Desired image size for the model input. Can be an integer for square images or a tuple (height, width) for specific dimensions. |
batch | int | 1 | Especifica o tamanho da inferência em lote do modelo de exportação ou o número máximo de imagens que o modelo exportado processará simultaneamente no modo predict. |
device | str | None | Especifica o dispositivo para exportação: CPU (device=cpu), MPS para Apple silicon (device=mps). |
Para mais detalhes sobre o processo de exportação, visite a página de documentação da Ultralytics sobre exportação.
Implementar Modelos TF GraphDef do YOLO26 Exportados
Depois de exportares o teu modelo YOLO26 para o formato TF GraphDef, o próximo passo é a implementação. O primeiro passo principal e recomendado para executar um modelo TF GraphDef é usar o método YOLO("model.pb"), conforme mostrado anteriormente no fragmento de código de utilização.
No entanto, para mais informações sobre como implementar os teus modelos TF GraphDef, consulta os seguintes recursos:
-
TensorFlow Serving: Um guia sobre o TensorFlow Serving que ensina como implementar e servir modelos de machine learning de forma eficiente em ambientes de produção.
-
TensorFlow Lite: Esta página descreve como converter modelos de machine learning num formato otimizado para inferência no dispositivo com o TensorFlow Lite.
-
TensorFlow.js: Um guia sobre conversão de modelos que ensina como converter modelos TensorFlow ou Keras para o formato TensorFlow.js para uso em aplicações web.
Resumo
Neste guia, explorámos como exportar modelos Ultralytics YOLO26 para o formato TF GraphDef. Ao fazeres isto, podes implementar de forma flexível os teus modelos YOLO26 otimizados em diferentes ambientes.
Para mais detalhes sobre a utilização, visita a documentação oficial do TF GraphDef.
Para mais informações sobre a integração do Ultralytics YOLO26 com outras plataformas e frameworks, vê a nossa página do guia de integração.
FAQ
Como posso exportar um modelo YOLO26 para o formato TF GraphDef?
Os modelos Ultralytics YOLO26 podem ser exportados para o formato TensorFlow GraphDef (TF GraphDef) de forma simples. Este formato fornece uma representação serializada e independente de plataforma do modelo, ideal para implementar em ambientes variados como dispositivos móveis e web. Para exportar um modelo YOLO26 para TF GraphDef, segue estes passos:
from ultralytics import YOLO
# Load the YOLO26 model
model = YOLO("yolo26n.pt")
# Export the model to TF GraphDef format
model.export(format="pb") # creates 'yolo26n.pb'
# Load the exported TF GraphDef model
tf_graphdef_model = YOLO("yolo26n.pb")
# Run inference
results = tf_graphdef_model("https://ultralytics.com/images/bus.jpg")Para mais informações sobre diferentes opções de exportação, visita a documentação da Ultralytics sobre exportação de modelos.
Quais são os benefícios de usar o TF GraphDef para a implementação do modelo YOLO26?
Exportar modelos YOLO26 para o formato TF GraphDef oferece múltiplas vantagens, incluindo:
- Independência de Plataforma: O TF GraphDef fornece um formato independente de plataforma, permitindo que os modelos sejam implementados em vários ambientes, incluindo dispositivos móveis e navegadores web.
- Otimizações: O formato permite várias otimizações, como 'constant folding', quantização e transformações de grafo, que melhoram a eficiência da execução e reduzem o uso de memória.
- Aceleração de Hardware: Os modelos no formato TF GraphDef podem tirar partido de aceleradores de hardware como GPUs, TPUs e chips de IA para ganhos de desempenho.
Lê mais sobre os benefícios na secção TF GraphDef da nossa documentação.
Por que devo usar o Ultralytics YOLO26 em vez de outros modelos de deteção de objetos?
O Ultralytics YOLO26 oferece inúmeras vantagens em comparação com outros modelos como o YOLOv5 e o YOLOv7. Alguns benefícios chave incluem:
- Desempenho de Ponta: O YOLO26 proporciona velocidade e precisão excecionais para deteção, segmentação e classificação de objetos em tempo real.
- Facilidade de Uso: Apresenta uma API intuitiva para treino, validação, previsão e exportação de modelos, tornando-o acessível tanto para principiantes como para especialistas.
- Ampla Compatibilidade: Suporta múltiplos formatos de exportação, incluindo ONNX, TensorRT, CoreML e TensorFlow, para opções de implementação versáteis.
Explora mais detalhes na nossa introdução ao YOLO26.
Como posso implementar um modelo YOLO26 em hardware especializado usando TF GraphDef?
Depois de um modelo YOLO26 ser exportado para o formato TF GraphDef, podes implementá-lo em várias plataformas de hardware especializado. Cenários típicos de implementação incluem:
- TensorFlow Serving: Usa o TensorFlow Serving para uma implementação de modelo escalável em ambientes de produção. Suporta gestão de modelos e serviço eficiente.
- Dispositivos Móveis: Converte modelos TF GraphDef para TensorFlow Lite, otimizado para dispositivos móveis e integrados, permitindo inferência no dispositivo.
- Navegadores Web: Implementa modelos usando TensorFlow.js para inferência do lado do cliente em aplicações web.
- Aceleradores de IA: Tira partido de TPUs e chips de IA personalizados para uma inferência acelerada.
Consulta a secção opções de implementação para informações detalhadas.
Onde posso encontrar soluções para problemas comuns ao exportar modelos YOLO26?
Para a resolução de problemas comuns com a exportação de modelos YOLO26, a Ultralytics fornece guias e recursos abrangentes. Se encontrares problemas durante a instalação ou exportação do modelo, consulta:
- Guia de Problemas Comuns: Oferece soluções para problemas frequentemente encontrados.
- Guia de Instalação: Instruções passo a passo para configurar os pacotes necessários.
Estes recursos devem ajudar-te a resolver a maioria dos problemas relacionados com a exportação e implementação do modelo YOLO26.