yolov8下实现绿萝识别

目录

一：背景

二：过程

---

一：背景

上一节我们学习了yolov8自带模型的使用，这一节我们讲解下yolov8的数据训练，生成模型来识别绿萝。

二：过程

1：数据准备，我们可以自己收集绿萝的图片，最起码需要准备几百张的图片。我们这里通过网络下载图片保存到一个目录等待处理。

2：数据标记

我们对绿萝图片进行标记，使用labelImg来进行标记。如果没有安装需要安装

pip install labelImg

安装成功在图片目录打开运行命令：labelImg

打开如下选择图片进行标记：

我这里可能标记的不太准确，正常一张图片可能绿萝比较小，可以较完整的标注每个绿萝。标记完会生成xml文件。里面包含了标记的坐标和名称等信息。这里需要注意的是可以设置yolo模式，这样生成txt文件后缀的。

3：处理标记文件

我们需要将xml文件处理为yolov8能识别的训练文件，如下代码：

import os
from os import listdir, getcwd
import re

classes = []  # 列表，储存标签类别
classes_dict = {}  # 字