additional class values

usdot-fhwa-OPS · Aug 21, 2024 · 9e0b3fd · 9e0b3fd
1 parent 4847aa7
commit 9e0b3fd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/v2i-hub/MUSTSensorDriverPlugin/src/MUSTSensorDetection.h b/src/v2i-hub/MUSTSensorDriverPlugin/src/MUSTSensorDetection.h
@@ -16,6 +16,9 @@ namespace MUSTSensorDriverPlugin {
     enum class DetectionClassification {
         SEDAN,
         TRUCK,
+        BUS,
+        PICKUP_TRUCK,
+        PEDESTRIAN,
         VAN,
         NA
     };
@@ -46,7 +49,10 @@ namespace MUSTSensorDriverPlugin {
     const static std::unordered_map<std::string, DetectionClassification> stringToDetectionClassificationMap = {
         {"sedan", DetectionClassification::SEDAN},
         {"truck", DetectionClassification::TRUCK},
-        {"van", DetectionClassification::VAN}
+        {"van", DetectionClassification::VAN},
+        {"bus", DetectionClassification::BUS},
+        {"pickup truck", DetectionClassification::PICKUP_TRUCK},
+        {"pedestrian", DetectionClassification::PEDESTRIAN}
     };
 
     /**