#EfficientNet

gen-efficientnet-pytorch:高效的EfficientNet PyTorch实现

2024年09月04日
Cover of gen-efficientnet-pytorch:高效的EfficientNet PyTorch实现
相关项目
Project Cover

gen-efficientnet-pytorch

本项目实现了EfficientNet、MixNet、MobileNetV3等多种高效模型,利用通用架构定义支持多种计算高效的神经网络。所有模型均基于MobileNet V1/V2块序列设计,并支持字符串化架构配置。请注意,该项目现已停止维护,推荐使用`timm`库获取更多功能和权重兼容的模型。

Project Cover

tf_efficientnet_b0.in1k

tf_efficientnet_b0.in1k是一个基于EfficientNet架构的图像分类模型,在ImageNet-1k数据集上使用Tensorflow训练,并由Ross Wightman移植到PyTorch。该模型具有5.3M参数和0.4 GMACs,支持细节丰富的224x224像素图像应用。其功能包括图像分类、特征映射提取和图像嵌入,非常适合在timm库中进行各种深度学习研究和应用,提供一种高效的图像处理方案。

Project Cover

efficientnet_b0.ra_in1k

efficientnet_b0.ra_in1k是基于ImageNet-1k数据集训练的EfficientNet-B0变体模型。它整合了RandAugment数据增强技术,采用RMSProp优化器和指数衰减学习率策略。模型拥有530万参数,0.4 GMACs计算量,适用于224x224像素输入。可应用于图像分类、特征提取和嵌入生成,为计算机视觉任务提供了平衡效率与性能的解决方案。

Project Cover

efficientnet_b3.ra2_in1k

efficientnet_b3.ra2_in1k是一款基于EfficientNet B3架构的图像分类模型,经过ImageNet-1k数据集训练。该模型采用RandAugment RA2数据增强策略,结合RMSProp优化器和指数衰减学习率调度,实现了优异的性能。拥有1220万参数,模型在图像分类、特征提取和嵌入生成等任务中表现出色,为计算机视觉应用提供了高效且灵活的解决方案。

Project Cover

efficientnet_lite0.ra_in1k

efficientnet_lite0.ra_in1k是一个在ImageNet-1k数据集上训练的轻量级图像分类模型。它使用RandAugment数据增强和RMSProp优化器,仅有4.7M参数和0.4 GMACs,适合资源受限环境。该模型支持图像分类、特征图提取和图像嵌入,通过timm库实现,提供简洁API接口,便于快速部署。

Project Cover

efficientnet-b0

EfficientNet是一种训练于ImageNet-1k数据集、分辨率为224x224的卷积模型。该模型提出了复合系数方法,以均衡缩放模型的深度、宽度和分辨率。在移动设备上表现卓越,适用于图像分类。同时,用户可在Hugging Face平台上获取特定任务的微调版本。

Project Cover

tf_efficientnet_b0.ns_jft_in1k

tf_efficientnet_b0.ns_jft_in1k模型基于EfficientNet架构,通过Noisy Student半监督学习在ImageNet-1k和JFT-300m数据集上训练。拥有520万参数,0.4 GMAC,支持224x224图像输入。这一轻量级模型适用于图像分类、特征提取和嵌入生成,为计算机视觉应用提供高效且多功能的解决方案。

Project Cover

efficientnet_b1.ra4_e3600_r240_in1k

efficientnet_b1.ra4_e3600_r240_in1k是基于EfficientNet架构的轻量级图像分类模型。该模型在ImageNet-1k数据集上训练,参数量为7.79M,在240x240输入尺寸下达到81.440%的Top-1准确率。它在模型大小和性能之间取得平衡,适用于计算资源有限的应用场景,也可作为特征提取器用于其他计算机视觉任务。

Project Cover

efficientnet_b4.ra2_in1k

efficientnet_b4.ra2_in1k是基于EfficientNet架构的图像分类模型,在ImageNet-1k数据集上训练。该模型采用RandAugment RA2增强方法,使用RMSProp优化器,具有1930万参数和3.1 GMACs计算量。支持320x320训练图像和384x384测试图像,可用于图像分类、特征提取和嵌入等任务,为计算机视觉领域提供高效解决方案。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号