efficientNe笔记

最新推荐文章于 2024-08-08 07:46:29 发布

原创最新推荐文章于 2024-08-08 07:46:29 发布 · 632 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#笔记

深度学习专栏收录该内容

1 篇文章

订阅专栏

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

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])

网络结构为：