Merge pull request meadx#27 from edwin-nz/master

Updated UNKNOWN condition to output to rospy.loginfo
edwin-nz · Dec 5, 2018 · 8f67594 · 8f67594
2 parents e02657a + 2022c2e
commit 8f67594
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ros/src/tl_detector/light_classification/tl_classifier.py b/ros/src/tl_detector/light_classification/tl_classifier.py
@@ -92,14 +92,18 @@ def get_classification(self, image):
 
             #print('Traffic Light Prediction ', predict[0])
             if predict[0][0] > self.threshold:
+                #print('Classifier Prediction - RED', predict[0][0])
                 rospy.loginfo("   Classifier Prediction - RED %f", predict[0][0])
                 return TrafficLight.RED
             elif predict[0][1] > self.threshold:
+                #print('Classifier Prediction - YELLOW', predict[0][1])
                 rospy.loginfo("   Classifier Prediction - YELLOW %f", predict[0][1])
                 return TrafficLight.YELLOW
             elif predict[0][2] > self.threshold:
+                #print('Classifier Prediction - YELLOW', predict[0][1])
                 rospy.loginfo("   Classifier Prediction - GREEN %f", predict[0][2])
                 return TrafficLight.GREEN
         else:
             #print('Classifier Prediction - UNKNOWN', label_vgg16[0][0])
+            rospy.loginfo("   Classifier Prediction - UNKNOWN [%s, %f]", label_vgg16[0][0][1], label_vgg16[0][0][2])
             return TrafficLight.UNKNOWN