Skip to content

Latest commit

 

History

History
25 lines (18 loc) · 1.29 KB

README.md

File metadata and controls

25 lines (18 loc) · 1.29 KB
Raw

yolov5_openvino_sdk

an SDK about how to use openvino model transformed from yolov5

Before Start

train your yolov5 model on your own dataset following Train Custom Data

export model to onnx，using export.py.

before run export.py:

  • changeto torch==1.5.1,torchvision==0.6.1
  • delete yolo.py, and rename yolo_1.py to yolo.py

then you can run export.py to export onnx model.

install OPENVINO2020R4

you can install following https://bbs.cvmart.net/topics/3117

Convert onnx to openvino

AS for how to use openvino to inference ,please refer to SDK using convert_openvino.sh to convert onnx to openvino.Then you can get openvino model .bin and .xml.

use openvino model to inference picture.

This profile provide an SDK. using ji.py to using transformed openvino model. (WARNING:NMS is not included in openvino model.)