Merge 25f8cd515bf66802766a17cfd0bd793981f0081a into 324671aa65785fd471920771de98baf9ca45ab43

This commit is contained in:
2022-11-03 15:23:22 +08:00 committed by GitHub
commit 48d71a5469
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 3 deletions

View 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
```

View File

@ -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()