YOLOv5 Segmentation Python

---

Run yolov5 segmentation model on onnxruntime or opencv dnn without torch!

Usage

All you need is on segment.py, it provides cli to run yolov5-seg onnx model.

python segment.py -- help

Image

python segment.py -m <YOLOV5-SEG-ONNX-PATH> -i <IMAGE-PATH>

Video

python segment.py -m <YOLOV5-SEG-ONNX-PATH> -v 0            # webcam
                                               <VIDEO-PATH> # local video

Note : Press q to stop video processing.

OpenCV DNN

Use Opencv DNN as backend with --dnn arguments.

python segment.py -m <YOLOV5-SEG-ONNX-PATH> -v 0 --dnn

Run on GPU

Auto using gpu to run model when devices is supported.

• onnxruntime need onnxruntime-gpu to be installed.

  Note : onnxruntime-gpu must be installed with the same version as onnxruntime to be able to use GPU.

• opencv-dnn need custom build.

pip install onnxruntime-gpu==<YOUR-ONNXRUNTIME-VERSION>

Reference

• https://github.com/ultralytics/yolov5
• https://github.com/UNeedCryDear/yolov5-seg-opencv-dnn-cpp