@kozmonos/image-classifier v0.0.5

Image Classification

This package allows you to interact with 3 simple functions (init, train, predict) using Tensorflow JS's @tensorflow-models/knn-classifier package.

const fs = require("fs");
const imageClassifier = require("@kozmonos/image-classifier");

// Initialize the classifier
const classifier = await imageClassifier.init();

// Train the classifier
const image = fs.readFileSync("./image.png");
const label = "label";
classifier.train(image, label);

const image2 = fs.readFileSync("./image2.png");
const label2 = "label2";
classifier.train(image2, label2);

// Predict the class of an image
const result = await classifier.predict(image);
/*
 out: {  classIndex: 0, label: 'label2',  confidences: { label2: 0.6666666666666666, label: 0.3333333333333333 } }
*/