YOLOV8 OBB 海思3516训练流程

YOLOV8 OBB 海思3516训练流程

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

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.0	

验证 ultralytics 是否安装好
import torch
torch.cuda.is_available()