Skip to content

vladmandic/tfjs-utils

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

38 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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