mobilenetv3_small_050.lamb_in1k

项目介绍：mobilenetv3_small_050.lamb_in1k

项目背景

Mobilenetv3_small_050.lamb_in1k 是一个基于移动设备优化的图像分类模型，由 timm 库提供并在 ImageNet-1k 数据集上进行训练。其创新的训练配方源于 ResNet 的强化策略，并通过 LAMB 优化算法进行了优化，适合应用在需要图像分类和特征提取的应用中。

模型详细信息

模型类型： 图像分类/特征骨干
模型统计：
- 参数数量（百万）：1.6
- GMACs：0.0
- 激活数（百万）：0.9
- 输入图像尺寸：224 x 224

训练配方

Mobilenetv3_small_050.lamb_in1k 的训练使用了一种改进的 LAMB 优化策略，与 ResNet 的某种 A2 配方类似。具体包括：

采用 LAMB 优化算法，较原始配方长 50% 的 EMA 权重平均，无需 CutMix。
使用 RMSProp（TensorFlow 1.0 实现方式）优化。
使用带有 warmup 的步骤（指数衰减且间歇性的）学习率调整。

数据集信息

此模型在大型数据集 ImageNet-1k 上进行训练，由此确保其在实际图像分类任务中表现卓越。

模型应用

图像分类

此模型能够对给定的图像进行分类，输出为图像中主要类别的概率分布。通过 timm 库，可以轻松加载预训练模型并进行推理。

特征图提取

Mobilenetv3_small_050.lamb_in1k 也可用于提取输入图像的特征图。其输出可用于进一步的视觉任务，如对象检测或图像分割。

图像嵌入

除了分类和特征提取外，用户还可以使用此模型生成图像嵌入。通过这种方式，可以通过倒数几层的输出，实现无监督的特征表示学习。

模型比较和引用

如果对该模型与其他模型的性能对比感兴趣，可以在 timm 的模型结果页面上进行探索。

如需引用此模型或相关研究，请参考以下文献：

@misc{rw2019timm,
  author = {Ross Wightman},
  title = {PyTorch Image Models},
  year = {2019},
  publisher = {GitHub},
  journal = {GitHub repository},
  doi = {10.5281/zenodo.4414861},
  howpublished = {\url{https://github.com/huggingface/pytorch-image-models}}
}

@inproceedings{howard2019searching,
  title={Searching for mobilenetv3},
  author={Howard, Andrew and Sandler, Mark and Chu, Grace and Chen, Liang-Chieh and Chen, Bo and Tan, Mingxing and Wang, Weijun and Zhu, Yukun and Pang, Ruoming and Vasudevan, Vijay and others},
  booktitle={Proceedings of the IEEE/CVF international conference on computer vision},
  pages={1314--1324},
  year={2019}
}