NYH 29af64039c V1.0 | 10 months ago | |
---|---|---|
models | 10 months ago | |
utils | 10 months ago | |
weights | 10 months ago | |
.dockerignore | 10 months ago | |
.gitattributes | 10 months ago | |
.gitignore | 10 months ago | |
Dockerfile | 10 months ago | |
LICENSE | 10 months ago | |
README.md | 10 months ago | |
detect.py | 10 months ago | |
hubconf.py | 10 months ago | |
lighting_app.py | 10 months ago | |
onnx_to_rknn.py | 10 months ago | |
requirements.txt | 10 months ago | |
rknn_detect_yolov5_0.py | 10 months ago | |
rknn_detect_yolov5_1.py | 10 months ago | |
rknn_detect_yolov5_best.py | 10 months ago | |
rknn_detect_yolov5_rtsp.py | 10 months ago | |
rknn_detect_yolov5_video.py | 10 months ago | |
test.py | 10 months ago | |
train.py | 10 months ago |
原版仓库: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