Fix class name when generating metrics for Object Detection

lmnet/lmnet/networks/object_detection/yolo_v1.py

@@ -98,6 +98,7 @@ def metrics(self, output, labels, thresholds=[0.3, 0.5, 0.7]):
         for overlap_thresh in thresholds:
             average_precisions = []
             for class_id, class_name in enumerate(self.classes):
+                class_name = class_name.replace(' ', '_')
                 tps = []
                 fps = []
                 scores = []

lmnet/lmnet/networks/object_detection/yolo_v2.py

@@ -248,6 +248,7 @@ def metrics(self, output, labels, thresholds=[0.3, 0.5, 0.7]):
             for overlap_thresh in thresholds:
                 average_precisions = []
                 for class_id, class_name in enumerate(self.classes):
+                    class_name = class_name.replace(' ', '_')
                     tps = []
                     fps = []
                     scores = []