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)
# 其它处理

检测结果：