efficientNe笔记

Thinkobj

于 2019-11-21 17:46:16 发布

阅读量610

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签：笔记

本文链接：https://blog.youkuaiyun.com/Thinkobj/article/details/103183244

深度学习专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍如何使用EfficientNet进行图像分类任务，包括安装、导入、加载预训练模型及特征提取等关键步骤。通过实例演示了EfficientNet在图像处理领域的高效应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Efficientnet 原文

使用pip安装efficientnet

pip install efficientnet_pytorch

导入efficientnet

from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_name('efficientnet-b0')  # load an EfficientNet
model = EfficientNet.from_pretrained('efficientnet-b0', num_classes=2)  # load a pretrained EfficientNet

EfficientNet 提取特征

from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained('efficientnet-b0')

# ... image preprocessing as in the classification example ...
print(img.shape) # torch.Size([1, 3, 224, 224])

features = model.extract_features(img)
print(features.shape) # torch.Size([1, 1280, 7, 7])

网络结构为：