mobilenet model can classify img. It provides simple interfaces to use. You can use your own category model to classify img.
npm install
npm run dev
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);
mobileNet:https://paddlejs.baidu.com/mobilenet