vgg16_worker.py 文件源码

def import_vgg16():
    global VGG16
    global Model
    global image
    global preprocess_input
    global K
    from keras.applications import VGG16
    from keras.models import Model
    from keras.preprocessing import image
    from keras.applications.vgg16 import preprocess_input

    from keras import backend as K
    if K.backend() == 'tensorflow':
        limit_mem()