三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

使用YOLO8训练一个自己的模型

使用YOLO8训练一个自己的模型

使用YOLO8训练一个自己的模型

目的,上一篇(使用YOLO8识别图片)调用模型进行了图像识别,识别图片中的小狗。

现在尝试训练自己的模型,然后调用自己的模型,比如使用小狗的图片训练自己的模型,然后调用这个模型识别图片中的小狗。

一、搜集图片

我简单下载了几张图片,供演示使用:

二、数据标注软件安装

可以选择LabelImg工具,这个适合新手。

另外还可以选择Label Studio(网页版,适合大批量  )和Roboflow(最省事)。

安装LabelImg

首先安装依赖库:

# 安装依赖
pip install PyQt5 lxml

← 返回列表