Skip to content

Repository of pre-trained YOLOv5 models in tensorflow.js format

License

Notifications You must be signed in to change notification settings

SkalskiP/yolov5js-zoo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

yolov5.js zoo

Logo

Convert

# clone YOLOv5 repository
git clone https://github.com/ultralytics/yolov5.git
cd yolov5

# create python virtual environment [recommended]
virtualenv venv
source venv/bin/activate

# install dependencies
pip install -r requirements.txt
pip install tensorflowjs

# convert model to tensorflow.js format
python export.py --weights yolov5s.pt --include tfjs

Deploy

Use YOLOv5.js npm package to run models on your website.

Contribute

Convert your YOLOv5 model to tensorflow.js, create pull request and allow others to run your AI in the browser.

Kudos

Kudos to ultralytics team as well as all other open-source contributors for building YOLOv5 project, and making it all possible.

License

Project is freely distributable under the terms of the MIT license.