项目介绍:tinynet_e.in1k
tinynet_e.in1k项目是一个图像分类模型,属于TinyNet家族。此模型是由论文作者经过在ImageNet-1k数据集上训练完成的。它的主要作用是对各种图像进行分类和特征提取。
模型详情
tinynet_e.in1k模型是一个图像分类和特征骨干模型。该模型拥有参数量为2.0M,激活数为0.7M,图片输入尺寸为106 x 106。相关研究论文是《Model rubik’s cube: Twisting resolution, depth and width for tinynets》,可以通过此链接查看。
模型统计
- 参数量 (Millions): 2.0
- GMACs: 0.0
- 激活 (Millions): 0.7
- 图像大小: 106 x 106
模型使用
图像分类
通过Python代码,使用timm库加载tinynet_e.in1k模型预训练版本,对输入图像执行分类操作。首先获取模型的具体转换步骤,比如归一化和图片大小调整,最后输出是图像的分类概率。
特征图提取
tinynet_e.in1k还可以用于提取特征图。通过去除分类层,可以提取输入图像的特征图。特征图会以多层卷积输出的格式呈现,例如,形状会是类似于torch.Size([1, 8, 53, 53])的张量。
图像嵌入
tinynet_e.in1k能够生成图像嵌入。其通过去除分类层,实现对图像的嵌入提取。处理后的输出可以作为其它下游任务的输入,例如进一步图像处理,或者其他机器学习模型的输入。
模型比较
tinynet_e.in1k模型的性能可以在timm库的模型结果页面进行比较,通过这些数据,我们能够更直观地了解tinynet_e.in1k相较于其他模型的优劣。
引用
要引用本模型的相关文献,可以使用下列BibTeX条目:
@article{han2020model,
title={Model rubik’s cube: Twisting resolution, depth and width for tinynets},
author={Han, Kai and Wang, Yunhe and Zhang, Qiulin and Zhang, Wei and Xu, Chunjing and Zhang, Tong},
journal={Advances in Neural Information Processing Systems},
volume={33},
pages={19353--19364},
year={2020}
}
@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}}
}
通过tinynet_e.in1k模型,可以更高效地处理图像分类任务,是一个高效的工具用于特征提取及图像嵌入生成,适合应用于各种需要轻量级模型的场景。