ONNX 模型库
返回模型

说明文档

模型卡片 - "鸟类物种分类器"

模型描述

"鸟类物种分类器"是一个先进的图像分类模型,旨在从图像中识别各种鸟类物种。它采用 EfficientNet 架构,并经过微调,在识别多种鸟类物种方面达到了很高的准确率。

如何使用

您可以使用以下代码在 Python 环境中轻松使用该模型:

from transformers import AutoFeatureExtractor, AutoModelForImageClassification

extractor = AutoFeatureExtractor.from_pretrained("chriamue/bird-species-classifier")
model = AutoModelForImageClassification.from_pretrained("chriamue/bird-species-classifier")

应用场景

  • 用于教育或生态研究的鸟类物种识别。
  • 协助生物多样性监测和保护工作。
  • 提升自然类应用和平台的用户体验。

训练数据

该模型基于"鸟类物种"数据集进行训练,这是一个全面的鸟类图像集合。该数据集的主要特征包括:

  • 物种总数:525 种鸟类。
  • 训练图像:84,635 张图像。
  • 验证图像:2,625 张图像。
  • 测试图像:2,625 张图像。
  • 图像格式:JPG 格式的彩色图像(224x224x3)。
  • 来源:来自 Kaggle。

训练结果

该模型经过 6 个 epoch 的训练后取得了令人印象深刻的结果:

  • 准确率:96.8%
  • 损失值:0.1379
  • 运行时长:136.81 秒
  • 每秒样本数:19.188
  • 每秒步数:1.206
  • 总训练步数:31,740

这些指标表明该模型具有很高的性能水平,使其在实际应用中具有可靠性。

局限性及偏见

  • 在不同的光照条件或图像质量下,模型的性能可能会有所不同。
  • 模型的准确率取决于训练数据集中的多样性和代表性。对于数据集中代表性不足的鸟类物种,该模型的效果可能较差。

伦理考量

应负责任地使用该模型,考虑隐私和环境影响。它不应被用于有害目的,例如针对濒危物种或违反野生动物保护法律。

致谢

我们要感谢 Kaggle 上数据集的创建者,是他们提供了丰富的数据源才使这个模型成为可能。

另请参阅

chriamue/bird-species-classifier

作者 chriamue

image-classification transformers
↓ 27.5K ♥ 22

创建时间: 2023-11-05 13:22:14+00:00

更新时间: 2023-11-12 10:37:45+00:00

在 Hugging Face 上查看

文件 (10)

.gitattributes
README.md
config.json
evaluate_model.py
model.onnx ONNX
model.safetensors
preprocessor_config.json
train.py
train_results.json
training_args.bin