Procházet zdrojové kódy

fix for arrange transforms

jiangjiajun před 5 roky
rodič
revize
57fcd37489
1 změnil soubory, kde provedl 16 přidání a 0 odebrání
  1. 16 0
      deploy/cpp/src/transforms.cpp

+ 16 - 0
deploy/cpp/src/transforms.cpp

@@ -197,6 +197,22 @@ void Transforms::Init(const YAML::Node& transforms_node, bool to_rgb) {
   to_rgb_ = to_rgb;
   for (const auto& item : transforms_node) {
     std::string name = item.begin()->first.as<std::string>();
+    if (name == "ArrangeClassifier") {
+      continue;
+    }
+    if (name == "ArrangeSegmenter") {
+      continue;
+    }
+    if (name == "ArrangeFasterRCNN") {
+      continue;
+    }
+    if (name == "ArrangeMaskRCNN") {
+      continue;
+    }
+    if (name == "ArrangeYOLOv3") {
+      continue;
+    }
+
     std::shared_ptr<Transform> transform = CreateTransform(name);
     transform->Init(item.begin()->second);
     transforms_.push_back(transform);