RK系列开发板模型转化加速代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
NYH 29af64039c V1.0 10 months ago
models V1.0 10 months ago
utils V1.0 10 months ago
weights V1.0 10 months ago
.dockerignore V1.0 10 months ago
.gitattributes V1.0 10 months ago
.gitignore V1.0 10 months ago
Dockerfile V1.0 10 months ago
LICENSE V1.0 10 months ago
README.md V1.0 10 months ago
detect.py V1.0 10 months ago
hubconf.py V1.0 10 months ago
lighting_app.py V1.0 10 months ago
onnx_to_rknn.py V1.0 10 months ago
requirements.txt V1.0 10 months ago
rknn_detect_yolov5_0.py V1.0 10 months ago
rknn_detect_yolov5_1.py V1.0 10 months ago
rknn_detect_yolov5_best.py V1.0 10 months ago
rknn_detect_yolov5_rtsp.py V1.0 10 months ago
rknn_detect_yolov5_video.py V1.0 10 months ago
test.py V1.0 10 months ago
train.py V1.0 10 months ago

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