非洲野生动物数据集#
Ultralytics 非洲野生动物数据集是一个目标检测数据集,包含 1,504 张图像,涵盖 4 个动物类别——水牛、大象、犀牛和斑马——这些动物常见于南非自然保护区。图像已预先划分为 1,052 张训练图像、225 张验证图像和 227 张测试图像,并且数据集会在你首次训练时自动下载(~100 MB)。这是一个紧凑、开箱即用的基准数据集,可用于训练用于野生动物监测、保护和生态研究的计算机视觉模型。
Watch: How to Train Ultralytics YOLO26 on the African Wildlife Dataset | Inference, Metrics & ONNX Export 🐘
数据集结构#
Ultralytics 非洲野生动物数据集包含 1,504 张图像,涵盖 4 个类别(水牛、大象、犀牛和斑马),并预先划分为三个子集:
- 训练集:1,052 张图像,每张图像都有对应的标注。
- 验证集:225 张图像,每张图像都有配套标注。
- 测试集:227 张图像,每张图像都有配套标注。
非洲野生动物数据集(~100 MB)会在你首次开始训练时自动下载,因此无需手动下载或准备。
探索 Ultralytics Platform 上的非洲野生动物数据集,浏览带有标注叠加层的图像,在 Charts 选项卡中查看类别分布和边界框热力图,并克隆数据集以在云端训练自己的模型。
应用#
Ultralytics 非洲野生动物数据集支持多种目标检测应用:
- 野生动物保护——检测并统计水牛、大象、犀牛和斑马,以支持自然保护区和保护地中的动物种群监测。
- 生态研究——研究不同栖息地中的物种分布和行为。
- 反盗猎监控——在大范围保护地的相机陷阱或无人机视频中标记动物。
- 教育和原型开发——用于学习模型训练和预测的紧凑型四类别数据集。
数据集 YAML#
YAML 文件定义数据集配置,包括路径、类别和其他相关详细信息。对于非洲野生动物数据集,african-wildlife.yaml 文件位于 https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/african-wildlife.yaml。
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# African Wildlife dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/detect/african-wildlife
# Example usage: yolo train data=african-wildlife.yaml
# parent
# ├── ultralytics
# └── datasets
# └── african-wildlife ← downloads here (100 MB)
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: african-wildlife # dataset root dir
train: images/train # train images (relative to 'path') 1052 images
val: images/val # val images (relative to 'path') 225 images
test: images/test # test images (relative to 'path') 227 images
# Classes
names:
0: buffalo
1: elephant
2: rhino
3: zebra
# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/african-wildlife.zip使用方法#
要在非洲野生动物数据集上训练 YOLO26n 模型 100 个周期,并将图像尺寸设为 640,请使用提供的代码示例。如需查看所有可用参数的完整列表,请参阅模型的训练页面。
from ultralytics import YOLO
# Load a model
model = YOLO("yolo26n.pt") # load a pretrained model (recommended for training)
# Train the model
results = model.train(data="african-wildlife.yaml", epochs=100, imgsz=640)训练完成后,使用微调后的模型对新图像运行推理:
from ultralytics import YOLO
# Load a model
model = YOLO("path/to/best.pt") # load an African wildlife fine-tuned model
# Inference using the model
results = model.predict("https://ultralytics.com/assets/african-wildlife-sample.jpg")示例图像和标注#
非洲野生动物数据集包含各种各样的图像,展示了多种动物及其自然栖息地。下面是数据集中的图像示例,每张图像都附有相应的标注。

- 拼接图像:这里展示了一个由拼接数据集图像组成的训练批次。拼接是一种训练技术,可将多张图像合并为一张,从而丰富批次多样性。这种方法有助于提升模型在不同目标尺寸、宽高比和上下文中的泛化能力。
引用、许可证和致谢#
我们感谢原始数据集作者 Bianca Ferreira 将此数据集发布给社区。Ultralytics 团队已在内部对其进行更新和改编,使其能够与 Ultralytics YOLO 模型无缝配合使用。源数据集未指定许可证。
如果你在研究中使用此数据集,请使用所述详细信息引用它:
@dataset{Ferreira_African_Wildlife_Ultralytics_Adaptation_2024,
author = {Ferreira, Bianca},
title = {African Wildlife Detection Dataset (Ultralytics YOLO Adaptation)},
url = {https://docs.ultralytics.com/datasets/detect/african-wildlife},
note = {Original dataset by Bianca Ferreira; adapted for Ultralytics YOLO by Glenn Jocher and Muhammad Rizwan Munawar},
license = {Not specified},
version = {1.0.0},
year = {2024}
}常见问题#
非洲野生动物数据集是一个目标检测数据集,包含 1,504 张图像,涵盖 4 个动物类别——水牛、大象、犀牛和斑马——这些动物生活在南非自然保护区。它用于训练和评估识别图像中非洲野生动物的模型,从而支持自然保护区中的野生动物保护、生态研究和监测。它也为学习计算机视觉的学生和研究人员提供了易于使用的资源。
Ultralytics 非洲野生动物数据集包含 1,504 张图像,涵盖 4 个类别:水牛、大象、犀牛和斑马。图像已预先划分为 1,052 张训练图像、225 张验证图像和 227 张测试图像,并且数据集会在你首次训练时自动下载(~100 MB)。
你可以使用
african-wildlife.yaml配置文件在非洲野生动物数据集上训练 YOLO26 模型。下面是使用图像尺寸 640 训练 YOLO26n 模型 100 个周期的示例:示例from ultralytics import YOLO # Load a model model = YOLO("yolo26n.pt") # load a pretrained model (recommended for training) # Train the model results = model.train(data="african-wildlife.yaml", epochs=100, imgsz=640)如需了解其他训练参数和选项,请参阅训练文档。
非洲野生动物数据集的 YAML 配置文件名为
african-wildlife.yaml,可在此 GitHub 链接中找到。此文件定义数据集配置,包括路径、类别以及训练机器学习模型所需的其他详细信息。源数据集未指定许可证。它最初由 Bianca Ferreira 发布在 Kaggle 上,后由 Ultralytics 进行改编,以便与 Ultralytics YOLO 模型无缝配合使用。如果你在研究中使用此数据集,请使用引用部分中的 BibTeX 条目引用它。



