diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py --- a/exp/kerokero/test.py +++ b/exp/kerokero/test.py @@ -24,7 +24,7 @@ model.summary() log.info("loading data...") with np.load(args.data) as data: - testImages=data["testImages"].reshape((-1,224,224,1)) + testImages=(np.float32(data["testImages"])/128-1).reshape((-1,224,224,1)) testLabels=data["testLabels"].reshape((-1,4,2)) log.info("done")