Updated localization-classification model

Generalized model to handle more lighting conditions.
SiliconCar · May 10, 2018 · 66a4928 · 66a4928
1 parent df0db26
commit 66a4928
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/ros/src/tl_detector/light_classification/ssd_kyle_v2/frozen_inference_graph.pb b/ros/src/tl_detector/light_classification/ssd_kyle_v2/frozen_inference_graph.pb
diff --git a/ros/src/tl_detector/light_classification/tl_classifier.py b/ros/src/tl_detector/light_classification/tl_classifier.py
@@ -65,7 +65,7 @@ def __init__(self, threshold, hw_ratio, sim_testing):
             # the tensorflow graph
 
         else: # site testing where we load the localization+classification model        
-            detect_model_name = 'ssd_kyle_v1'
+            detect_model_name = 'ssd_kyle_v2'
             PATH_TO_CKPT = detect_model_name + '/frozen_inference_graph.pb'
             # setup tensorflow graph
             self.detection_graph = tf.Graph()