我想使用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文件中。但这个错误不断出现。
1条答案
按热度按时间a6b3iqyw1#
“序列长度为零”表示数据加载器没有提供数据。这通常是由于提供的路径有问题,
flow_from_directory
会静默失败。一种确认的方法是@Idavid在他们的评论中建议的,但无论哪种方法,您都可以尝试this thread中建议的任何选项。