以RKNN系列模型为例,训练一个yolov7的目标检测模型
使用RKNN系列模型训练YOLOv7目标检测模型: 数据标注: 使用Labelimg标注图片,修改predefined_classes.txt。 模型训练: SSH进入计算服务器Docker,按README进行训练。将标注数据放入yolov7-rknn/datasets,修改数据集配置文件。在yolov7-rknn文件夹下执行train.py训练。训练后,使用export.py导出ONNX文件。 模型转换与部署: 参考airockchip/rknn-toolkit2进行模型转换。将转换好的模型拷贝至部署代码的model/RK3588文件夹,编写类别信息。修改include/postprocess.h和src/postprocess.cc,重新编译代码。按install.md文件执行,并使用orangepi-config使能串口。
July 8, 2025 • 1 min read
目标检测 RKNN