如何在AWS Lambda中使用Keras VGG16模型?

时间:2019-11-26 03:27:41

标签: keras deployment aws-lambda transfer-learning vgg-net

我正在尝试在aws lambda上使用keras vgg16模型进行预测。但是我得到了一个错误:尝试下载模型时没有空间了。

from keras.applications.imagenet_utils import preprocess_input
from keras.applications import VGG16, nasnet, xception, inception_v3
from keras.preprocessing import image
from keras.applications.imagenet_utils import preprocess_input
from keras.models import Model
import json

def loadmodel(use_model):
    m_model = VGG16(weights = 'imagenet')
    last_layer = Model(inputs=m_model.input, 
    outputs=m_model.get_layer("fc2").output)
    return m_model_sans_last_layer

model_load=loadmodel("m_VGG16")

我得到了错误:

[Errno 28] No space left on device

我正在使用3000 mb lambda内存。

如何在AWS Lambda上使用该模型?

我可以保存最后一层并仅加载最后一层进行预测吗?

0 个答案:

没有答案