1.准备环境
pytorch-1.8.1,torchvision-0.9.1,python-3.9,
然后安装ultralytics。
pip install ultralytics
2.准备数据集+配置文件
在训练YOLOv8时,选择的数据格式是VOC,然后将数据集转换成可以直接让YOLOv8进行使用。
(数据集转换格式请参照此篇文章DIOR数据集xml转txt格式并划分训练集测试集验证集(用于yolo)-CSDN博客)
配置文件dior.yaml
3.下载权重
下载对应版本的模型https://github.com/ultralytics/assets/releases