diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py --- a/exp/kerokero/test.py +++ b/exp/kerokero/test.py @@ -3,6 +3,7 @@ import logging as log import numpy as np from keras.models import load_model +from keras.applications.inception_v3 import preprocess_input from prepare_data import loadDataset,Sample from analyzer.epoint import EPoint @@ -16,23 +17,22 @@ parser.add_argument("data") args=parser.parse_args() model=load_model(args.model) +model.summary() log.info("loading data...") with np.load(args.data) as data: - trainImages=data["trainImages"] - trainLabels=data["trainLabels"] testImages=data["testImages"] testLabels=data["testLabels"] log.info("done") -log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels)) +log.info(model.evaluate(preprocess_input(testImages).reshape((-1,224,224,3)),testLabels)) for img in testImages: - label=model.predict(np.reshape(img,(1,224,224,1))) + label=model.predict(preprocess_input(np.reshape(img,(1,224,224,3)))) print(label) points=[] for i in range(4): points.append(EPoint((label[0][i*2]+1)*112,(label[0][i*2+1]+1)*112)) corners=Corners(points) - sample=Sample(np.uint8(img*255),corners) + sample=Sample(img,corners) sample.show()