Merge 25f8cd515bf66802766a17cfd0bd793981f0081a into 324671aa65785fd471920771de98baf9ca45ab43
This commit is contained in:
commit
48d71a5469
15
examples/onnx/yolov5/README.md
Normal file
15
examples/onnx/yolov5/README.md
Normal file
@ -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
|
||||
```
|
||||
|
||||
@ -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()
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user