Modelos TensorFlow.js

Explore modelos pré-treinados para adicionar visão computacional, processamento de linguagem natural (PNL) e outras tarefas comuns de ML aos seus aplicativos da Web e baseados em navegador.

Visão

Analise recursos em imagens e vídeos. Desbloqueie novas experiências em tempo real no navegador.

Classificação de imagens

Classifique imagens com rótulos do banco de dados ImageNet (MobileNet).

Detecção de objetos

Localize e identifique vários objetos em uma única imagem (Coco SSD).

Segmentação semântica

Execute a segmentação semântica no navegador (DeepLab).

Corpo

Detecte pontos-chave e poses no rosto, mãos e corpo com modelos do MediaPipe e outros, otimizados para JavaScript e Node.js.

Detecção de rosto simples

Detecte rostos em imagens usando uma arquitetura Single Shot Detector com um codificador personalizado (Blazeface).

Detecção de pontos de referência faciais

Preveja 486 pontos de referência faciais 3D para inferir a geometria aproximada da superfície de rostos humanos.

Detecção de pose

API unificada de detecção de pose para usar um dos três modelos que ajudam a detectar poses atípicas e movimentos corporais rápidos com desempenho em tempo real.

Segmentação corporal

Segmente pessoas e partes do corpo em tempo real.

Detecção de pose de mão

Detector de palma e modelo de rastreamento de dedo do esqueleto da mão. Preveja 21 pontos-chave de mão 3D por mão detectada.

Estimativa de profundidade de retrato

Estime um mapa de profundidade para uma única imagem de retrato de um ser humano.

Texto

Habilite a PNL em seu aplicativo da web usando o poder do BERT e outras arquiteturas de codificador Transformer.

Resposta a perguntas em linguagem natural

Responda perguntas com base no conteúdo de uma determinada passagem do texto usando BERT.

Detecção de toxicidade de texto

Pontue o impacto percebido que um comentário pode ter em uma conversa, de “Muito tóxico” a “Muito saudável” (Toxicidade).

Codificador de frase universal

Codifique texto em incorporações para tarefas de PNL, como classificação de sentimentos e similaridade textual (Universal Sentence Encoder).

Áudio

Classifique o áudio para detectar sons e acionar uma ação em seu aplicativo web.

Reconhecimento de comando de fala

Classifique trechos de áudio de 1 segundo do conjunto de dados de comandos de fala (comandos de fala).

Em geral

Encontre mais modelos do TensorFlow.js que podem ser usados ​​imediatamente.

Classificador KNN

Utilitário para criar um classificador usando o algoritmo K-Nearest-Neighbors. Pode ser usado para aprendizagem por transferência.