Object Detection using YOLOv5 and Tensorflow.js

Object Detection application right in your browser. Serving YOLOv5 in browser using tensorflow.js with webgl backend.

Setup

git clone https://github.com/Hyuto/yolov5-tfjs.git
cd yolov5-tfjs
yarn install #Install dependencies

Scripts

yarn start # Start dev server
yarn build # Build for productions

Model

YOLOv5n model converted to tensorflow.js.

used model : yolov5n
size       : 7.5 Mb

Use another model

Use another YOLOv5 model.

Clone yolov5 repository

git clone https://github.com/ultralytics/yolov5.git && cd yolov5

Install requirements.txt first

pip install -r requirements.txt

Export model to tensorflow.js format

export.py --weights yolov5*.pt --include tfjs

Update modelName in App.jsx to new model name

...
// model configs
const modelName = "yolov5*"; // change to new model name
const classThreshold = 0.25;
...