White list only for manual config

hookttg · May 4, 2022 · eaaac7a · eaaac7a
1 parent 3e2ac15
commit eaaac7a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/example/examples.h b/example/examples.h
@@ -621,6 +621,13 @@ class YoloDarknetExample final : public VideoExample
             config.emplace("maxBatch", std::to_string(maxBatch));
             config.emplace("classNames", pathToModel + "coco.names");
             config.emplace("maxCropRatio", "-1");
+
+			config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_person));
+			config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_car));
+			config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_bicycle));
+			config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_motorbike));
+			config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_bus));
+			config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_truck));
         }
         else
         {
@@ -635,13 +642,6 @@ class YoloDarknetExample final : public VideoExample
             config.emplace("inference_precison", m_trackerSettings.m_inferencePrecison);
         }
 
-        config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_person));
-        config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_car));
-        config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_bicycle));
-        config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_motorbike));
-        config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_bus));
-        config.emplace("white_list", std::to_string((objtype_t)ObjectTypes::obj_truck));
-
         m_detector = BaseDetector::CreateDetector(tracking::Detectors::Yolo_Darknet, config, frame);
         if (m_detector.get())
         {