tanlam1703 / tfjs-yolov5-example

TensorFlow.js example for YOLOv5

tfjs-yolov5-example

Demo

A YOLOv5 tfjs demo is on https://zldrobit.github.io/tfjs-yolov5-example/. You could drag a file to the center box to detect objects with a pretrained COCO model.

Usage

Export TensorFlow.js model as described in ultralytics/yolov5#1127

After the release of YOLOv5 v6, users could export tfjs models with its main branch:

git clone https://github.com/ultralytics/yolov5.git
python export.py --weights yolov5s.pt --include tfjs

Local Test

After exported the tfjs model, clone this repo:

git clone https://github.com/zldrobit/tfjs-yolov5-example.git
cd tfjs-yolov5-example

Install packages with npm:

npm install

Link YOLOv5 weights directory into public directory:

ln -s ../../yolov5/yolov5s_web_model public/web_model

Run YOLOv5 detection web service with:

npm start

Github Pages Depolyment

Edit the homepage and deploy fields in package.json changing

"homepage": "https://zldrobit.github.io/tfjs-yolov5-example",

to

"homepage": "https://GITHUB_USERNAME.github.io/tfjs-yolov5-example",

and

"deploy": "gh-pages -d build --repo [email protected]:zldrobit/tfjs-yolov5-example.git"

to

"deploy": "gh-pages -d build --repo [email protected]:GITHUB_USERNAME/tfjs-yolov5-example.git"

Run

npm run deploy

PS: This repo assumes the model input resolution is 640x640. If you change the --img value in exporting *.pb, change modelWidth and modelHeight in src/index.js accordingly.

EDIT:

• Add github pages deployment support
• Use GitHub project site (previously use GitHub User Page)

Reference

https://medium.com/hackernoon/tensorflow-js-real-time-object-detection-in-10-lines-of-code-baf15dfb95b2