Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Pretrained models for TensorFlow.js

License

NotificationsYou must be signed in to change notification settings

tensorflow/tfjs-models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This repository hosts a set of pre-trained models that have been ported toTensorFlow.js.

The models are hosted on NPM and unpkg so they can be used in any project out of the box. They can be used directly or used in a transfer learningsetting with TensorFlow.js.

To find out about APIs for models, look at the README in each of the respectivedirectories. In general, we try to hide tensors so the API can be used bynon-machine learning experts.

For those interested in contributing a model, please file aGitHub issue on tfjs to gaugeinterest. We are trying to add models that complement the existing set of modelsand can be used as building blocks in other apps.

Models

TypeModelDemoDetailsInstall
Images
MobileNet
liveClassify images with labels from theImageNet database.npm i @tensorflow-models/mobilenet
source
Hand
liveReal-time hand pose detection in the browser using TensorFlow.js.npm i @tensorflow-models/hand-pose-detection
source
Pose
liveAn API for real-time human pose detection in the browser.npm i @tensorflow-models/pose-detection
source
Coco SSD
Object detection model that aims to localize and identify multiple objects in a single image. Based on theTensorFlow object detection API.npm i @tensorflow-models/coco-ssd
source
DeepLab v3
Semantic segmentationnpm i @tensorflow-models/deeplab
source
Face Landmark Detection
liveReal-time 3D facial landmarks detection to infer the approximate surface geometry of a human facenpm i @tensorflow-models/face-landmarks-detection
source
Audio
Speech Commands
liveClassify 1 second audio snippets from thespeech commands dataset.npm i @tensorflow-models/speech-commands
source
Text
Universal Sentence Encoder
Encode text into a 512-dimensional embedding to be used as inputs to natural language processing tasks such as sentiment classification and textual similarity.npm i @tensorflow-models/universal-sentence-encoder
source
Text Toxicity
liveScore the perceived impact a comment might have on a conversation, from "Very toxic" to "Very healthy".npm i @tensorflow-models/toxicity
source
Depth Estimation
Portrait Depth
liveEstimate per-pixel depth (the distance to the camera center) for a single portrait image, which can be further used for creative applications such as3D photo andrelighting.npm i @tensorflow-models/depth-estimation
source
General Utilities
KNN Classifier
This package provides a utility for creating a classifier using the K-Nearest Neighbors algorithm. Can be used for transfer learning.npm i @tensorflow-models/knn-classifier
source

Development

You can run the unit tests for any of the models by running the followinginside a directory:

yarn test

New models should have a test NPM script (seethispackage.json andrun_tests.tshelper for reference).

To run all of the tests, you can run the following command from the root of thisrepo:

yarn presubmit

[8]ページ先頭
©2009-2025 Movatter.jp