Skip to content

Testing out general object detection on TensorFlow.JS using YOLOv8-Nano

License

Notifications You must be signed in to change notification settings

donjuanpond/tfjs-yolov8n

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Software License PRs Welcome

TensorFlow.js: YOLOv8n test

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 YOLOv8n 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/. The model is trained off the COCO dataset, so its just for general object detection.

This model loads and runs far faster than its contemporaries and is very small.

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

About

Testing out general object detection on TensorFlow.JS using YOLOv8-Nano

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published