使用YOLO8训练一个自己的模型
使用YOLO8训练一个自己的模型
目的,上一篇(使用YOLO8识别图片)调用模型进行了图像识别,识别图片中的小狗。
现在尝试训练自己的模型,然后调用自己的模型,比如使用小狗的图片训练自己的模型,然后调用这个模型识别图片中的小狗。
一、搜集图片
我简单下载了几张图片,供演示使用:
二、数据标注软件安装
可以选择LabelImg工具,这个适合新手。
另外还可以选择Label Studio(网页版,适合大批量 )和Roboflow(最省事)。
安装LabelImg
首先安装依赖库:
# 安装依赖
pip install PyQt5 lxml
