niyouhao
/
RK3588_Detection
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
RK系列开发板模型转化加速代码
2 Commits 1 Branch 0 Tags 126 KiB
Python 98.9%
Dockerfile 1%
Shell 0.1%
Go to file
NYH 29af64039c V1.0 2023-12-27 14:00:44 +08:00
models V1.0 2023-12-27 14:00:44 +08:00
utils V1.0 2023-12-27 14:00:44 +08:00
weights V1.0 2023-12-27 14:00:44 +08:00
.dockerignore V1.0 2023-12-27 14:00:44 +08:00
.gitattributes V1.0 2023-12-27 14:00:44 +08:00
.gitignore V1.0 2023-12-27 14:00:44 +08:00
Dockerfile V1.0 2023-12-27 14:00:44 +08:00
LICENSE V1.0 2023-12-27 14:00:44 +08:00
README.md V1.0 2023-12-27 14:00:44 +08:00
detect.py V1.0 2023-12-27 14:00:44 +08:00
hubconf.py V1.0 2023-12-27 14:00:44 +08:00
lighting_app.py V1.0 2023-12-27 14:00:44 +08:00
onnx_to_rknn.py V1.0 2023-12-27 14:00:44 +08:00
requirements.txt V1.0 2023-12-27 14:00:44 +08:00
rknn_detect_yolov5_0.py V1.0 2023-12-27 14:00:44 +08:00
rknn_detect_yolov5_1.py V1.0 2023-12-27 14:00:44 +08:00
rknn_detect_yolov5_best.py V1.0 2023-12-27 14:00:44 +08:00
rknn_detect_yolov5_rtsp.py V1.0 2023-12-27 14:00:44 +08:00
rknn_detect_yolov5_video.py V1.0 2023-12-27 14:00:44 +08:00
test.py V1.0 2023-12-27 14:00:44 +08:00
train.py V1.0 2023-12-27 14:00:44 +08:00

README.md

原版仓库：https://github.com/ultralytics/yolov5

环境要求：python version >= 3.6

模型训练：python3 train.py

模型导出：python3 models/export.py --weights "xxx.pt"

转换rknn：python3 onnx_to_rknn.py

模型推理：python3 rknn_detect_yolov5.py