vladmandic / tfjs-utils

Collection of generic JS utils when working with TF models

Home Page:https://vladmandic.github.io/tfjs-utils/src/kernels.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tfjs utils

collection of generic js utils when working with tf models

tools

  • signature.js: analyze saved model or graph model input/output tensors based on either model signature or model executor
  • classifier.js: generic tf image classifier
  • detector.js: generic tf object detector
  • freeze.py: convert tf frozen model to saved model
  • tfbun.js: list/search models on <tfhub.com>
  • kernels.js: generate html table of supported tfjs kernels per backend
  • tfinfo.py: get cuda & gpu info

classes

/classes/*.json:
contains class label definitions for different datasets in JSON format:

  • CoCo
  • ImageNet 1k
  • OpenImages v4
  • DeepDetect
  • Places365

/classes/wordnet-synset.json.gz:
full wordnet word definitions that can be used to parse word IDs included in individual classes definitions

About

Collection of generic JS utils when working with TF models

https://vladmandic.github.io/tfjs-utils/src/kernels.html

License:MIT License


Languages

Language:JavaScript 74.0%Language:HTML 15.3%Language:Python 5.4%Language:TypeScript 5.4%