Teachable Machine Node
Teachable Machine Node empowers you to load any image classification model trained with Google's Teachable Machine tool in a Node.Js project.
Teachable Machine makes AI easy for everyone, by offering a fast and fun way to train a real TensorFlow.js Machine Learning Models without any coding required. You can train the computer to recognize images, sounds, & poses, using your camera or your own dataset. Check The Awesome Teachable Machine List full of useful resources and amazing projects to gain some cool ideas.
For now, Teachable Machine Node holds support only for image models, but we won't stop here. Check out the Roadmap of what comes next!
- Install using npm
npm install @sashido/teachablemachine-node
- Install using yarn
yarn add @sashido/teachablemachine-node
Create you own Model with Teachable Machine
Gathering samples is the fundamental first step to your Teachable Machine Model. Use your camera to collect data or upload some preselected images.
Advanced option for further insights on the model performance and accuracy. Once certain it returns valid results, hit the Export option.
- Make sure that you select Tensorflow.js format when exporting. That way your model will be uploaded (for free) and you will receive an access URL.
- Pass the URL to the teachablemachine-node to load the model. Next, let it have the image URL and call
classifyto get the predictions.
Play around with our pre-trained 'Is It A Dog?' model.
SashiDo's team is full of notorious animal lovers and no wonder the sample model was trained to recognize dog pics from other images. 😊 We've collected a dataset of more than 2000 images of dogs, cats, horses, other animals, people and everyday objects and uploaded them into two different classes with Teachable Machine.
A big shoutout to Unsplash.com as a great platform for free photos. Check it out in case you're just starting and do not have raw data yet.
You can load our Is_It_A_Dog model using this URL and following the example below:
Here's a quick example of how to load the model in your project.
const TeachableMachine = ;const model =modelUrl: "";model;
And the alternative for all Express fans.
const express = ;const TeachableMachine = ;const model =modelUrl: "";const app = ;const port = 3000;app;app;
In the long run, we will add more options, so you can train and load all kinds of Teachable Machine Models.
Add support for Pose Models.
Add support for Audio Models.
Add support for Gifs.
Add support for Videos.
We would love to have your opinion which's the one you would like to see supported first. Don't be shy and drop us a line at firstname.lastname@example.org.
Contributors of any kind are welcome. Share your awesome improvements in a pull request and join our mission to make Machine Learning more affordable & accessible!
Copyright © 2020, CloudStrap AD. See LICENSE for further details.