Я разрабатываю приложение на Android, для которого мне нужны черты лица снятого изображения на телефоне Android. VGG - 16 соответствует моим потребностям. Я просто хочу сделать связь между Android и моделью Keras, чтобы я мог получить функции после block5_pool (MaxPooling2D) в Android. Размер импульса составляет (1,224,224,3), а выходной - (1,7,7,512). Я хочу использовать те же веса Imagenet для VGG.
from keras.applications.vgg16 import VGG16
from keras.applications.vgg16 import preprocess_input
model = VGG16(weights='imagenet', include_top=False)
model.summary()
img = image.load_img(img_path, target_size=(224, 224))
img_data = image.img_to_array(img)
img_data = np.expand_dims(img_data, axis=0)
img_data = preprocess_input(img_data)
vgg16_feature = model.predict(img_data)
Это vgg16_feature - это то, что мне нужно в Android для каждого захваченного изображения пользователем.