Avoid implicitly copying a matrix to numpy_image.

5b32723f · Davis King · fff97234 · 5b32723f
Commit 5b32723f authored May 27, 2018 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

numpy_image.h dlib/python/numpy_image.h +4 -2

No files found.
--- a/dlib/python/numpy_image.h
+++ b/dlib/python/numpy_image.h
@@ -188,15 +188,17 @@ namespace dlib
            return *this;
        }

+        template <long NR, long NC>
        numpy_image (
-            matrix<pixel_type>&& rhs
+            matrix<pixel_type,NR,NC>&& rhs
        )
        {
            *this = convert_to_numpy(std::move(rhs));
        }

+        template <long NR, long NC>
        numpy_image& operator= (
-            matrix<pixel_type>&& rhs
+            matrix<pixel_type,NR,NC>&& rhs
        )
        {
            *this = convert_to_numpy(std::move(rhs));