keras ValueError:尝试使用MobileNet获取影像要素时,要求检索元素0,但序列长度为0

8iwquhpp  于 2023-01-26  发布在  其他
关注(0)|答案(1)|浏览(268)

我想使用mobileNet来获取数据集中存在的所有图像的特征。但是提到的错误不断弹出。
产生错误的完整代码为:
'

from keras.applications import MobileNet
from keras.preprocessing.image import ImageDataGenerator
from keras.models import Model
from keras.applications.mobilenet import preprocess_input

# Load the MobileNet model
base_model = MobileNet(weights='imagenet', include_top=False)

# Define the data generator
data_generator = ImageDataGenerator(preprocessing_function=preprocess_input)

# Create the data generator for the custom dataset
image_generator = data_generator.flow_from_directory(
        'C:/Users/mahit/OneDrive/Pictures/Desktop/DATA AND VIDEOS//16k_images',
        target_size=(224, 224),
        batch_size=1,
        class_mode=None,
        shuffle=False)

# Obtain the image features
image_features = base_model.predict(image_generator, 16042)

# Save the image features to a file
np.save('mobilenet_image_features.npy', image_features)

'
完整错误如下所示:

我想获取图像特征并将其保存在一个.npy文件中。但这个错误不断出现。

a6b3iqyw

a6b3iqyw1#

“序列长度为零”表示数据加载器没有提供数据。这通常是由于提供的路径有问题,flow_from_directory会静默失败。
一种确认的方法是@Idavid在他们的评论中建议的,但无论哪种方法,您都可以尝试this thread中建议的任何选项。

相关问题