search

linhaoqi027 / yolov5_openvino_sdk

an SDK about how to use openvino model transformed from yolov5
36 stars 11 forks source link

NMS #2

Open linhaoqi027 opened 4 years ago

linhaoqi027 commented 4 years ago

NMS is not included in openvino model. My solution is transform the output of openvino to tensor. Then using NMS in yolov5 to generate output.This may not a perfect solution. You can use NMS functon in openvino repository . And if you can dao this, contract with me.