keras 读取模型进行测试的方式

from keras.preprocessing import image
import numpy as np
from keras.models import load_model
import os
from keras.applications.resnet50 import preprocess_input
from shutil import copyfile
from keras.preprocessing.image import ImageDataGenerator

work_dir = 'E:/pcb_image_data/data_small'
#test_data_dir = 'E:/pcb_image_data/data_small/test'
#载入模型
def read_model():
    model = load_model(work_dir + '/model_weight.h5')
    return model

#读取多张图片
def read_img_array(img_dir):
    img = []
    for f in os.listdir(img_dir):
        image_path = os.path.join(img_dir, f)
        if os.path.isfile(image_path):
            images = image.load_img(image_path, target_size=(100, 100))
            x = image.img_to_array(images)
            x = np.expand_dims(x, axis=0)
            img.append(x)
    x = np.concatenate([x for x in img])
    
    #读取模型进行预测
    model = load_model(work_dir + '/model_weight.h5'