fix: bbox clamping in torch inference

jolibrain · Jan 28, 2022 · 2d6efd3 · 2d6efd3
1 parent 02872eb
commit 2d6efd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
@@ -1587,7 +1587,7 @@ namespace dd
                         bbox[2]
                             = std::min(static_cast<double>(cols - 1), bbox[2]);
                         bbox[3]
-                            = std::min(static_cast<double>(cols - 1), bbox[3]);
+                            = std::min(static_cast<double>(rows - 1), bbox[3]);
 
                         APIData ad_bbox;
                         ad_bbox.add("xmin", bbox[0]);