collection of generic js utils when working with tf models
signature.js
: analyze saved model or graph model input/output tensors based on either model signature or model executorclassifier.js
: generic tf image classifierdetector.js
: generic tf object detectorfreeze.py
: convert tf frozen model to saved modeltfbun.js
: list/search models on <tfhub.com>kernels.js
: generate html table of supported tfjs kernels per backendtfinfo.py
: get cuda & gpu info
/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