# import the necessary packages from config import tiny_imagenet_config as config from pyimagesearch.preprocessing import imagetoarraypreprocessor as ITA from pyimagesearch.preprocessing import simplespreprocessor as SP from pyimagesearch.preprocessing import meanpreprocessor as MP from pyimagesearch.io import hdf5datasetgenerator as HDFG from pyimagesearch.utils.ranked import rank5_accuracy from keras.models import load_model import json # load the RGB means for the training set means = json.loads(open(config.DATASET_MEAN).read()) # initialize the image preprocessors sp = SP.SimplePreprocessor(64,64) mp = MP.MeanPreprocessor(means['R'],means['G'],means['B']) iap = ITA.ImageToArrayPreprocessor() # initialize the testing dataset generator testGen = HDFG.HDF5DatasetGenerator(config.TEST_HDF5, 64,preprocessors=[sp, mp, iap], classes=config.NUM_CLASSES) # load the pre-trained network print("[INFO] loading model...") model = load_model(config.MODEL_PATH) # make predictions on the testing data print("[INFO] predicting on test data...") predictions = model.predict_generator(testGen.generator(),steps=testGen.numImages // 64, max_queue_size=64 * 2) # compute the rank-1 and rank-5 accuracies
from pyimagesearch.preprocessing import simplespreprocessor as SP from pyimagesearch.preprocessing import patchpreprocessor as PP from pyimagesearch.preprocessing import meanpreprocessor as MP from pyimagesearch.preprocessing import croppreprocessor as CP from pyimagesearch.io import hdf5datasetgenerator as HDF from pyimagesearch.utils.ranked import rank5_accuracy from keras.models import load_model import numpy as np import progressbar import json # 加载RGB均值数据 means = json.loads(open(config.DATASET_MEAN).read()) # 初始化预处理 sp = SP.SimplePreprocessor(227, 227) mp = MP.MeanPreprocessor(means['R'], means['G'], means['B']) cp = CP.CropPreprocessor(227, 227) iap = IAP.ImageToArrayPreprocessor() # 加载训练好的模型 print("[INFO] loading model ...") model = load_model(config.MODEL_PATH) # 初始化测试数据集生成器,并进行预测 print("[INFO] predicting on test data (no crops)...") testGen = HDF.HDF5DatasetGenerator(config.TEST_HDF5, 64, preprocessors=[sp, mp, iap], classes=2) predictions = model.predict_generator(testGen.generator(),