GitHub - ninetailskim/yolov5_for_rknn: YOLOv5 in PyTorch > ONNX > RKNN

环境要求：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

Name		Name	Last commit message	Last commit date
Latest commit History 893 Commits
.github		.github
data		data
models		models
utils		utils
weights		weights
.dockerignore		.dockerignore
.gitattributes		.gitattributes
.gitignore		.gitignore
Dockerfile		Dockerfile
LICENSE		LICENSE
README.md		README.md
detect.py		detect.py
hubconf.py		hubconf.py
onnx_to_rknn.py		onnx_to_rknn.py
requirements.txt		requirements.txt
rknn_detect_yolov5.py		rknn_detect_yolov5.py
test.py		test.py
train.py		train.py

Provide feedback