diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py
--- a/exp/kerokero/test.py
+++ b/exp/kerokero/test.py
@@ -3,13 +3,12 @@ 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
+from k_util import averageDistance,generateData
 import config as cfg
 
 keras.losses.averageDistance=averageDistance
@@ -25,11 +24,11 @@ model.summary()
 
 log.info("loading data...")
 with np.load(args.data) as data:
-	testImages=data["testImages"]
-	testLabels=data["testLabels"]
+	testImages=data["testImages"].reshape((-1,224,224,1))
+	testLabels=data["testLabels"].reshape((-1,4,2))
 log.info("done")
 
-log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels.reshape((-1,4,2))))
+log.info(model.evaluate(testImages,testLabels))
 
 for img in testImages:
 	label=model.predict(np.reshape(img,(1,224,224,1)))