上篇地址:yolo图像识别教程 - 以原神人物识别为例(3)-CSDN博客
本篇主要内容为图像标注工具导出及yolov8的安装
一、图像标注
上篇介绍了YOLO V8的label标签的格式,现在介绍如何使用X-AnyLabeling-CPU版导出
(一)编制类别的txt文件后导出
例如:
需要注意的是标签需要为英文,不同类别之间换行。编制好后,点击上方的导出yolo标签后选中该txt后即可导出。默认导出为label文件夹。
二、导入YOLO V8
这里先提供yolov8官方文档链接Home - Ultralytics YOLOv8 Docs
YOLO V8默认提供两种安装方式,分别pip与源代码安装。本文使用pip安装方式。
pip install ultralytics
导入成功后,即可载入官方自定义模型,需要注意的是,此api会自动下载模型(需要科学上网),网络条件不好的,可以自行下载,放在py文件的同级目录下。也可配置yaml文件(后文会介绍),自定义模型。
model = YOLO("yolov8m.pt") #预训练模型
model = YOLO('yolov8n.yaml') #自定义模型
之后会介绍如何训练自己的图形识别模型,来识别原神人物。