diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py --- a/exp/kerokero/test.py +++ b/exp/kerokero/test.py @@ -3,7 +3,6 @@ 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 @@ -25,14 +24,14 @@ with np.load(args.data) as data: testLabels=data["testLabels"] log.info("done") -log.info(model.evaluate(preprocess_input(testImages).reshape((-1,224,224,3)),testLabels)) +log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels)) for img in testImages: - label=model.predict(preprocess_input(np.reshape(img,(1,224,224,3)))) + label=model.predict(np.reshape(img,(1,224,224,1))) 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(img,corners) + sample=Sample(np.uint8((img+1)*128),corners) sample.show()