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.
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
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
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 git@github.com:zldrobit/tfjs-yolov5-example.git"
to
"deploy": "gh-pages -d build --repo git@github.com: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)
https://github.com/zldrobit/tfjs-yolov5-example/wiki/FAQ
Add yolov5 in the web using yolov5js: https://github.com/SkalskiP/yolov5js