ONNX 模型库
返回模型

说明文档

Coop Cam 鸟类分类器

Int8 量化 MobileNetV2,用于 ESP32-S3 设备端的鸟类物种分类。

模型详情

  • 架构: MobileNetV2 (width_mult=0.25)
  • 输入: 96x96 灰度图 (int8),形状 [1, 96, 96, 1]
  • 输出: 168 种北美鸟类 (int8)
  • 大小: 612 KB (TFLite int8)
  • 验证集准确率: 58.8% (量化后)

训练

使用 chriamue/bird-species-dataset 数据集,从 chriamue/bird-species-classifier (EfficientNet-B2, 525 类) 进行知识蒸馏。

  • 蒸馏温度: 4.0
  • Alpha (KL 权重): 0.7
  • 优化器: AdamW (lr=1e-3, weight_decay=1e-4)
  • 学习率调度器: 余弦退火,50 个 epoch
  • 早停: 耐心值 10

文件

文件 描述
Int8 量化 TFLite 模型 (部署到 ESP32)
C 数组头文件,用于嵌入固件
C 头文件,包含物种名称数组
PyTorch 检查点 (用于重新训练)
ONNX 中间文件 (用于重新导出)

使用方法

对于 ESP32-S3 固件,在构建中包含 和 。

要从 TFLite 文件重新生成 :

许可证

MIT (源自 MIT 许可的教师模型和 CC0 数据集)

mvdmakesthings/coopcam-bird-classifier

作者 mvdmakesthings

↓ 1 ♥ 0

创建时间: 2026-02-14 16:26:53+00:00

更新时间: 2026-02-14 16:27:29+00:00

在 Hugging Face 上查看

文件 (7)

.gitattributes
README.md
best_student.pt
bird_classifier.tflite
model_data.h
species_labels.h
student.onnx ONNX