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