diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py --- a/exp/kerokero/test.py +++ b/exp/kerokero/test.py @@ -3,12 +3,17 @@ import logging as log import numpy as np from keras.models import load_model +import keras.losses +import keras.metrics from prepare_data import loadDataset,Sample from analyzer.epoint import EPoint from analyzer.corners import Corners +from k_util import averageDistance import config as cfg +keras.losses.averageDistance=averageDistance +keras.metrics.averageDistance=averageDistance parser=argparse.ArgumentParser() parser.add_argument("model") @@ -24,14 +29,14 @@ with np.load(args.data) as data: testLabels=data["testLabels"] log.info("done") -log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels)) +log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels.reshape((-1,4,2)))) for img in testImages: 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)) + points.append(EPoint((label[0][i][0]+1)*112,(label[0][i][1]+1)*112)) corners=Corners(points) sample=Sample(np.uint8((img+1)*128),corners) sample.show()