fix: fix read color config

ichiro-its · Feb 21, 2024 · 37a250b · 37a250b
1 parent 5c894b0
commit 37a250b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/data/color_classifier.json b/data/color_classifier.json
@@ -0,0 +1,14 @@
+{
+  "ball": {
+    "min_hsv": [151, 40, 10],
+    "max_hsv": [179, 100, 100]
+  },
+  "field": {
+    "min_hsv": [84, 10, 0],
+    "max_hsv": [116, 100, 100]
+  },
+  "line": {
+    "min_hsv": [180, 0, 55],
+    "max_hsv": [260, 60, 100]
+  }
+}
diff --git a/src/ninshiki_cpp/detector/color_detector.cpp b/src/ninshiki_cpp/detector/color_detector.cpp
@@ -62,9 +62,10 @@ bool ColorDetector::load_configuration(const std::string & path)
   nlohmann::json config = nlohmann::json::parse(input);
   for (auto & item : config.items()) {
     // Get all config
+    std::string name = item.key();
     try {
       utils::Color color(
-        item.value().at("name"),
+        name,
         item.value().at("min_hsv")[0],
         item.value().at("max_hsv")[0],
         item.value().at("min_hsv")[1],