From 7029a28084b46dc128663b091e02e9bf913f0804 Mon Sep 17 00:00:00 2001 From: Bill Lotter Date: Sun, 28 Aug 2016 23:00:10 -0400 Subject: [PATCH] Added PredNet to custom_objects --- kitti_evaluate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kitti_evaluate.py b/kitti_evaluate.py index bde1a3a..b74d4eb 100644 --- a/kitti_evaluate.py +++ b/kitti_evaluate.py @@ -20,7 +20,7 @@ from kitti_settings import * -n_plot = 20 +n_plot = 40 batch_size = 10 nt = 10 @@ -33,7 +33,7 @@ f = open(json_file, 'r') json_string = f.read() f.close() -train_model = model_from_json(json_string) +train_model = model_from_json(json_string, custom_objects = {'PredNet': PredNet}) train_model.load_weights(weights_file) # Create testing model (to output predictions)