ssd-mobilenet-coco-tensorflowjs-model

Tensorflow.js pretrained model of ssd_mobilenet_v1_coco

How to use it?

Clone this repository

git clone https://github.com/DavidCai1993/ssd-mobilenet-coco-tensorflowjs-model.git

Load model.json in your tensorflow.js application

'use strict'
const tf = require('@tensorflow/tfjs')
require('@tensorflow/tfjs-node')

;(async function () {
  const model = await tf.loadModel(`file://${__dirname}/ssd-mobilenet-coco-tensorflowjs-model/model/model.json`)

  // ...
})(console.error)