Saltar para o conteúdo

Exportar para o formato de modelo TF.js a partir de um formato de modelo YOLO11

A implantação de modelos de aprendizado de máquina diretamente no navegador ou no Node.js pode ser complicada. Terá de se certificar de que o formato do modelo está optimizado para um desempenho mais rápido, de modo a que o modelo possa ser utilizado para executar aplicações interactivas localmente no dispositivo do utilizador. O formato de modelo TensorFlow.js, ou TF.js, foi concebido para utilizar o mínimo de energia e proporcionar um desempenho rápido.

A funcionalidade "exportar para o formato de modelo TF.js" permite-lhe otimizar os seus Ultralytics YOLO11 modelos para inferência de deteção de objetos em alta velocidade e executada localmente. Neste guia, vamos orientá-lo na conversão de seus modelos para o formato TF.js, facilitando o bom desempenho dos modelos em vários navegadores locais e aplicativos Node.js.

Por que razão deve exportar para TF.js?

Exportar os seus modelos de aprendizagem automática para TensorFlow.js, desenvolvido pela equipa TensorFlow como parte do ecossistema TensorFlow mais vasto, oferece inúmeras vantagens para a implementação de aplicações de aprendizagem automática. Ajuda a melhorar a privacidade e a segurança do utilizador, mantendo os dados sensíveis no dispositivo. A imagem abaixo mostra a arquitetura do TensorFlow.js e a forma como os modelos de aprendizagem automática são convertidos e implementados nos navegadores Web e no Node.js.

TFArquitetura .js

A execução de modelos localmente também reduz a latência e proporciona uma experiência de utilizador mais reactiva. TensorFlowO .js também vem com capacidades offline, permitindo que os utilizadores utilizem a sua aplicação mesmo sem uma ligação à Internet. TF O .js foi concebido para a execução eficiente de modelos complexos em dispositivos com recursos limitados, uma vez que foi concebido para ser escalável, com suporte de aceleração GPU .

Principais caraterísticas do TF.js

Eis as principais caraterísticas que fazem do TF.js uma ferramenta poderosa para os programadores:

  • Suporte multiplataforma: TensorFlow.js pode ser utilizado em ambientes de browser e Node.js, proporcionando flexibilidade na implementação em diferentes plataformas. Permite que os programadores criem e implementem aplicações mais facilmente.

  • Suporte para vários back-ends: TensorFlow.js suporta vários back-ends para computação, incluindo CPU, WebGL para aceleração GPU , WebAssembly (WASM) para velocidade de execução quase nativa e WebGPU para capacidades avançadas de aprendizagem automática baseadas no browser.

  • Capacidades offline: Com o TensorFlow.js, os modelos podem ser executados no browser sem necessidade de uma ligação à Internet, tornando possível o desenvolvimento de aplicações que funcionam offline.

Opções de implantação com TensorFlow.js

Antes de nos debruçarmos sobre o processo de exportação de modelos YOLO11 para o formato TF.js, vamos explorar alguns cenários de implementação típicos em que este formato é utilizado.

TFO .js fornece uma gama de opções para implementar os seus modelos de aprendizagem automática:

  • Aplicações de ML no navegador: É possível criar aplicações Web que executam modelos de aprendizagem automática diretamente no browser. A necessidade de computação do lado do servidor é eliminada e a carga do servidor é reduzida.

  • Aplicações Node.js:: TensorFlow O .js também suporta a implementação em ambientes Node.js, permitindo o desenvolvimento de aplicações de aprendizagem automática do lado do servidor. É particularmente útil para aplicações que requerem o poder de processamento de um servidor ou o acesso a dados do lado do servidor.

  • Extensões do Chrome: Um cenário de implantação interessante é a criação de extensões do Chrome com TensorFlow.js. Por exemplo, pode desenvolver uma extensão que permita aos utilizadores clicar com o botão direito do rato numa imagem em qualquer página Web para a classificar utilizando um modelo de ML pré-treinado. TensorFlow O .js pode ser integrado nas experiências quotidianas de navegação na Web para fornecer informações imediatas ou aumentos baseados na aprendizagem automática.

Exportação de modelos YOLO11 para TensorFlow.js

Pode expandir a compatibilidade de modelos e a flexibilidade de implementação convertendo os modelos YOLO11 para TF.js.

Instalação

Para instalar o pacote necessário, execute:

Instalação

# Install the required package for YOLO11
pip install ultralytics

Para obter instruções detalhadas e melhores práticas relacionadas com o processo de instalação, consulte o nosso guia de instalaçãoUltralytics . Ao instalar os pacotes necessários para YOLO11, se encontrar alguma dificuldade, consulte o nosso guia de Problemas comuns para obter soluções e dicas.

Utilização

Antes de se debruçar sobre as instruções de utilização, é importante notar que, embora todos os modelos deUltralytics YOLO11 estejam disponíveis para exportação, pode certificar-se de que o modelo selecionado suporta a funcionalidade de exportação aqui.

Utilização

from ultralytics import YOLO

# Load the YOLO11 model
model = YOLO("yolo11n.pt")

# Export the model to TF.js format
model.export(format="tfjs")  # creates '/yolo11n_web_model'

# Load the exported TF.js model
tfjs_model = YOLO("./yolo11n_web_model")

# Run inference
results = tfjs_model("https://ultralytics.com/images/bus.jpg")
# Export a YOLO11n PyTorch model to TF.js format
yolo export model=yolo11n.pt format=tfjs  # creates '/yolo11n_web_model'

# Run inference with the exported model
yolo predict model='./yolo11n_web_model' source='https://ultralytics.com/images/bus.jpg'

Para obter mais detalhes sobre as opções de exportação suportadas, visite a página de documentaçãoUltralytics sobre opções de implantação.

Implantação de modelos exportados YOLO11 TensorFlow .js

Agora que você exportou o modelo YOLO11 para o formato TF.js, a próxima etapa é implantá-lo. A primeira etapa principal e recomendada para executar um TF.js é usar o YOLO("./yolo11n_web_model") como mostrado anteriormente no trecho de código de uso.

No entanto, para obter instruções detalhadas sobre a implantação dos modelos TF.js, consulte os seguintes recursos:

Resumo

Neste guia, aprendemos como exportar modelos Ultralytics YOLO11 para o formato TensorFlow.js. Ao exportar para TF.js, você ganha a flexibilidade de otimizar, implantar e dimensionar seus modelos YOLO11 em uma ampla variedade de plataformas.

Para mais pormenores sobre a utilização, visite a documentação oficialTensorFlow.js.

Para obter mais informações sobre a integração do Ultralytics YOLO11 com outras plataformas e estruturas, não se esqueça de conferir nossa página de guia de integração. Ela está repleta de ótimos recursos para ajudá-lo a aproveitar ao máximo o YOLO11 em seus projetos.

FAQ

Como é que exporto os modelos Ultralytics YOLO11 para o formato TensorFlow.js?

A exportação dos modelos Ultralytics YOLO11 para o formato TensorFlow.js (TF.js) é simples. Pode seguir estes passos:

Utilização

from ultralytics import YOLO

# Load the YOLO11 model
model = YOLO("yolo11n.pt")

# Export the model to TF.js format
model.export(format="tfjs")  # creates '/yolo11n_web_model'

# Load the exported TF.js model
tfjs_model = YOLO("./yolo11n_web_model")

# Run inference
results = tfjs_model("https://ultralytics.com/images/bus.jpg")
# Export a YOLO11n PyTorch model to TF.js format
yolo export model=yolo11n.pt format=tfjs  # creates '/yolo11n_web_model'

# Run inference with the exported model
yolo predict model='./yolo11n_web_model' source='https://ultralytics.com/images/bus.jpg'

Para obter mais detalhes sobre as opções de exportação suportadas, visite a página de documentaçãoUltralytics sobre opções de implantação.

Por que razão devo exportar os meus modelos YOLO11 para TensorFlow.js?

A exportação de modelos YOLO11 para TensorFlow.js oferece várias vantagens, incluindo:

  1. Execução local: Os modelos podem ser executados diretamente no browser ou no Node.js, reduzindo a latência e melhorando a experiência do utilizador.
  2. Suporte multiplataforma: TF.js suporta vários ambientes, permitindo flexibilidade na implantação.
  3. Capacidades offline: Permite que as aplicações funcionem sem uma ligação à Internet, garantindo fiabilidade e privacidade.
  4. GPU Aceleração: Aproveita o WebGL para a aceleração GPU , optimizando o desempenho em dispositivos com recursos limitados.

Para obter uma visão geral abrangente, consulte as nossas Integrações com TensorFlow.js.

Como é que o TensorFlow.js beneficia as aplicações de aprendizagem automática baseadas no browser?

TensorFlow.js foi especificamente concebido para a execução eficiente de modelos de ML em browsers e ambientes Node.js. Eis como beneficia as aplicações baseadas no browser:

  • Reduz a latência: Executa modelos de aprendizagem automática localmente, fornecendo resultados imediatos sem depender de cálculos do lado do servidor.
  • Melhora a privacidade: Mantém os dados sensíveis no dispositivo do utilizador, minimizando os riscos de segurança.
  • Permite a utilização offline: Os modelos podem funcionar sem uma ligação à Internet, garantindo uma funcionalidade consistente.
  • Suporta vários backends: Oferece flexibilidade com backends como CPU, WebGL, WebAssembly (WASM) e WebGPU para necessidades computacionais variadas.

Interessado em saber mais sobre TF.js? Consulte o guia oficial do TensorFlow.js.

Quais são as principais caraterísticas do TensorFlow.js para a implantação de modelos YOLO11 ?

As principais caraterísticas do TensorFlow.js incluem:

  • Suporte multiplataforma: TF.js pode ser utilizado tanto em navegadores Web como em Node.js, proporcionando uma grande flexibilidade de implementação.
  • Vários backends: Suporta CPU, WebGL para aceleração GPU , WebAssembly (WASM) e WebGPU para operações avançadas.
  • Capacidades offline: Os modelos podem ser executados diretamente no browser sem ligação à Internet, o que o torna ideal para o desenvolvimento de aplicações Web com capacidade de resposta.

Para cenários de implantação e informações mais detalhadas, consulte a nossa secção sobre Opções de implantação com TensorFlow.js.

Posso implementar um modelo YOLO11 em aplicações Node.js do lado do servidor utilizando TensorFlow.js?

Sim, o TensorFlow.js permite a implementação de modelos YOLO11 em ambientes Node.js. Isto permite aplicações de aprendizagem automática do lado do servidor que beneficiam do poder de processamento de um servidor e do acesso a dados do lado do servidor. Os casos de utilização típicos incluem processamento de dados em tempo real e pipelines de aprendizagem automática em servidores backend.

Para começar a utilizar a implantação do Node.js, consulte o guia Executar TensorFlow.js no Node.js em TensorFlow.

📅C riado há 7 meses ✏️ Atualizado há 1 mês

Comentários