@paddlejs-models/mobilenet NPM

mobilenet

mobilenet model can classify img. It provides simple interfaces to use. You can use your own category model to classify img.

Run Demo

Execute in the current directory

npm install
npm run dev

Visit http://0.0.0.0:8867

Usage

import * as mobilenet from '@paddlejs-models/mobilenet';

// You need to specify your model path and the binary file count
// If your has mean and std params, you need to specify them.
// map is the results your model can classify.
await mobilenet.load({
    path,
    mean: [0.485, 0.456, 0.406],
    std: [0.229, 0.224, 0.225]
}, map);

// get the result the mobilenet model classified.
const res = await mobilenet.classify(img);