Made max_cost_assignment() give an error if you incorrectly give it a

non-square matrix.
daben · Jun 18, 2015 · c9c3fa1 · c9c3fa1
1 parent 3286a0a
commit c9c3fa1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/python/src/other.cpp b/tools/python/src/other.cpp
@@ -57,6 +57,9 @@ list _max_cost_assignment (
     const matrix<double>& cost
 )
 {
+    if (cost.nr() != cost.nc())
+        throw dlib::error("The input matrix must be square.");
+
     // max_cost_assignment() only works with integer matrices, so convert from
     // double to integer.
     const double scale = (std::numeric_limits<dlib::int64>::max()/1000)/max(abs(cost));