YOLOV8 OBB 海思3516训练流程

YOLOV8 OBB 海思3516训练流程

目录
1、 下载带GPU版本的torch（可选） 1
2、 安装 ultralytics 2
3、 下载pycharm 社区版 2
4、安装pycharm 3
5、新建pycharm 工程 3
6、 添加conda 环境 4
7、 训练代码 5
9、配置Ymal 文件 6
10、修改网络结构 9
11、运行train.py 开始训练模型 10
12、检查数据是否标注正确 11

1、下载带GPU版本的torch（可选）

在pytorch官网：
https://pytorch.org/get-started/locally/
下载GPU版本的torch。

`pip3 install torch torchvision torchaudio --index-url` https://download.pytorch.org/whl/cu118

2、安装 ultralytics

安装ultralytics
运行：

conda create -n yolov8 python==3.8 -y
conda activate yolov8
pip install ultralytics==8.0.5
pip install onnx==1.12