fix: raise exception when a bbox file contains invalid classes

jolibrain · Feb 14, 2023 · 3a82a9d · 3a82a9d
1 parent 6db03be
commit 3a82a9d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/backends/torch/torchdataset.cc b/src/backends/torch/torchdataset.cc
@@ -871,6 +871,12 @@ namespace dd
         std::string val;
         iss >> val;
         int cls = std::stoi(val);
+        if (cls <= 0)
+          {
+            throw InputConnectorBadParamException(
+                "Dataset contains an invalid class:" + std::to_string(cls)
+                + " in file " + bboxfname);
+          }
         classes.push_back(target_to_tensor(cls));
 
         std::vector<double> bbox(4);