Skip to main content
100%

Using TensorFlow.js with MobileNet models for image classification on Node.js

✓ Published0🌍 Public
TTheLady
Last edited Jul 26, 2018
Created on Jul 4, 2019

This example shows image classification inference running in Node.js using TensorFlow.js and a MobileNet model. It predicts categories for a JPEG input file by decoding the image with jpeg-js, converting pixel data into an int32 tensor via tf.tensor3d, and passing it to mobilenet.MobileNet’s classify method. The script loads the model from a local file path and prints the classification results to the console. It demonstrates how to bridge image decoding and tensor creation for server-side machine learning workflows.

AI-generated description
MIT Licensed

Similar vizzes