若该文为原创文章,转载请注明原文出处。
本文基于rknn_model_zoo和山水无移大佬的博客和代码训练模型并部署到正点原子的ATK-DLRK3568板子测试。
https://github.com/ultralytics/ultralytics
一、训练
1、环境搭建
使用的是AUTODL环境,yolov8-obb数据集不大,也可以使用cpu。
2、创建虚拟环境
# 创建虚拟环境
conda create -n yolov8_env python=3.8
# 初始化
source activate
# 激活
conda activate yolov8_env
3、下载yolov8源码
https://github.com/ultralytics/ultralytics
可以使用git克隆,也可以直接下载zip文件,拷贝进入在解压。
发现yolov8源码无法在通过源码安装了
4、安装
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
5、训练yolov8-obb
执行命令开始训练
yolo obb train data=dota8.yaml