YOLOv8 训练15种动物分类模型

.sout

已于 2024-09-02 09:34:23 修改

阅读量104

点赞数 1

分类专栏： yolo 文章标签： YOLO 目标检测机器学习

于 2024-08-30 15:54:46 首次发布

本文链接：https://blog.csdn.net/m0_52604918/article/details/141720023

版权

yolo 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

1. 下载数据集

https://hyper.ai/datasets/31084

2. 进行训练

这里数据集没做拆分，训练测试都是同一个数据集。

train.py:

```python
from ultralytics import YOLO
from ultralytics.utils import DEFAULT_CFG
from datetime import datetime

current_time = datetime.now()
time_str = current_time.strftime("%Y-%m-%d_%H-%M-%S") 
DEFAULT_CFG.save_dir = f"./models/{time_str}"

if __name__ == "__main__":
    model = YOLO("yolov8x-cls.pt")

    # animal_data下放三个文件夹，分别为train、test、val
    results = model.train(data="./animal_data/", epochs=200, imgsz=224, device=0, save=True, save_period=1 ,batch=16)

4.检测

代码：

# 加载训练好的模型
model = YOLO('best.pt')
 

results = model(['./u950468431,735343220fm253fmtauto.jpg', './15-animals-data/u2189055899,1855606441fm253fmtau.png'], save=True)
# 其它处理