TensorFlow.js: Digit Detection

This code is built off Sascha Dittman's object detection with TensorFlow.JS demo: https://github.com/SaschaDittmann/tfjs-cv-objectdetection. The model is a custom-trained YOLOv8 model, converted to the TensorFlow.JS format with this tutorial: https://docs.ultralytics.com/modes/export/#arguments. The code to convert raw model output into coherent bounding boxes comes from https://github.com/Hyuto/yolov8-tfjs/. Also, big shout-out to netron.app for making life much easier.

Setup

Prepare the node environments:

$ yarn

Run the local web server script:

$ node server.js

Then, to get a URL to the port this has ran the server on, install localtunnel and use it:

$ npm install -g localtunnel
$ lt --port 3000