TensorFlow 2.6运行EfficientNetV2时出现AttributeError问题求助

问题根源 #

TensorFlow 2.6版本的keras.applications模块未内置EfficientNetV2模型，该模型在TensorFlow 2.9及更高版本才被正式纳入官方内置应用模块。

可行解决方案 #

方案1：升级TensorFlow版本 #

将tensorflow-gpu升级至2.9或以上版本，即可直接使用官方内置的EfficientNetV2：

pip install --upgrade tensorflow-gpu>=2.9

升级后可通过以下代码调用模型：

from tensorflow.keras.applications import efficientnet_v2
# 示例：加载预训练的EfficientNetV2B0模型
model = efficientnet_v2.EfficientNetV2B0(weights='imagenet')

方案2：不升级TensorFlow，安装第三方实现包 #

如果无法升级TensorFlow版本，可以安装第三方的EfficientNetV2实现库：

pip install efficientnet-v2

安装后通过以下代码调用模型：

from efficientnet_v2 import EfficientNetV2B0
model = EfficientNetV2B0(weights='imagenet')

内容的提问来源于stack exchange，提问作者Cy Rine