Skip to content

Latest commit

 

History

History
executable file
·
54 lines (32 loc) · 1.08 KB

README.md

File metadata and controls

executable file
·
54 lines (32 loc) · 1.08 KB

YOLOP

Input

Input

(Image from https://github.com/hustvl/YOLOP/blob/main/inference/images/0ace96c3-48481887.jpg)

Ailia input shape: (1, 3, 1280, 720) Range: [-1.0, 1.0]

Output

Output

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample image,

$ python3 yolop.py

If you want to specify the input image, put the image path after the --source option. You can use --save-dir option to change the name of the output file to save.

$ python3 yolop.py  --source inference/videos --save-dir inference/output

It can also be used for video files. As with images, you can use them by specifying the path with the --source option.

$ python3 yolop.py --source inference/videos

Reference

YOLOP

Model Format

ONNX opset = 11

Framework

Pytorch

Netron

yolop.onnx.prototxt