diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py --- a/exp/kerokero/test.py +++ b/exp/kerokero/test.py @@ -1,4 +1,5 @@ import argparse +import logging as log import numpy as np from keras.models import load_model @@ -6,6 +7,7 @@ from keras.models import load_model from prepare_data import loadDataset,Sample from analyzer.epoint import EPoint from analyzer.corners import Corners +import config as cfg parser=argparse.ArgumentParser() @@ -15,13 +17,15 @@ args=parser.parse_args() model=load_model(args.model) -print("loading data...") +log.info("loading data...") with np.load(args.data) as data: trainImages=data["trainImages"] trainLabels=data["trainLabels"] testImages=data["testImages"] testLabels=data["testLabels"] -print("done") +log.info("done") + +log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels)) for img in testImages: label=model.predict(np.reshape(img,(1,224,224,1)))