Added PredNet to custom_objects

RenieWell · Aug 29, 2016 · 7029a28 · 7029a28
1 parent 7728ca1
commit 7029a28
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)