Modelos do TensorFlow.js

Descubra modelos pré-treinados para adicionar visão computacional, processamento de linguagem natural (PLN) e outras tarefas comuns de ML aos seus aplicativos baseados no navegador e na Web.

Visão

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

Classificação de imagens

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

Detecção de objetos

Localize e identifique diversos objetos em uma única imagem (Coco SSD).

Segmentação semântica

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

Corpo

Detecte os principais pontos e poses em rostos, mãos e corpos com modelos do MediaPipe e muito mais, otimizados para JavaScript e Node.js.

Detecção facial simples

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

Detecção de pontos de referência do rosto

Previsão de 486 pontos de referência em 3D no rosto para inferência da geometria aproximada da superfície dos rostos humanos.

Detecção de poses

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

Segmentação do corpo

Segmente pessoas e partes do corpo em tempo real.

Detecção da posição das mãos

Modelo de detecção de palmas e rastreamento de dedos de mão de esqueleto. Previsão de 21 pontos principais em 3D em cada mão detectada.

Estimativa de profundidade de retrato

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

Texto

Ative o PLN no app da Web usando o poder do BERT e de outras arquiteturas de codificadores do Transformer.

Resposta a perguntas em linguagem natural

Responda a perguntas com base no conteúdo de um trecho de texto usando BERT.

Detecção de toxicidade em textos

Classifique o impacto percebido de um comentário sobre uma conversa de "Muito tóxico" a "Muito saudável" (Toxicity).

Codificador de frases universais

Codifique texto em embeddings para tarefas de PLN, como classificação de sentimento e similaridade textual (Universal Sentence Encoder).

Áudio

Classifique áudios para detectar sons e acionar uma ação no app da Web.

Reconhecimento de comandos por voz

Classificar trechos de áudio de um segundo em um conjunto de comandos de voz (speech-commands).

Geral

Encontre mais modelos do TensorFlow.js prontos para uso.

Classificador KNN

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