diff --git a/examples/onnx/yolov5/README.md b/examples/onnx/yolov5/README.md new file mode 100644 index 0000000..cb664c6 --- /dev/null +++ b/examples/onnx/yolov5/README.md @@ -0,0 +1,15 @@ +# YOLOV5 DEMO + +### 1.About onnx file + +- This model remove the post-process layer(which is not available on NPU now). +- To export .onnx format model file, refer to https://github.com/airockchip/rknn_model_zoo/tree/main/models/vision/object_detection/yolov5-pytorch + + + +### 2.Run demo + +``` +python test.py +``` + diff --git a/examples/onnx/yolov5/test.py b/examples/onnx/yolov5/test.py index a1c9988..13066f1 100644 --- a/examples/onnx/yolov5/test.py +++ b/examples/onnx/yolov5/test.py @@ -310,8 +310,8 @@ if __name__ == '__main__': if boxes is not None: draw(img_1, boxes, scores, classes) # show output - # cv2.imshow("post process result", img_1) - # cv2.waitKey(0) - # cv2.destroyAllWindows() + cv2.imshow("post process result", img_1) + cv2.waitKey(0) + cv2.destroyAllWindows() rknn.release()