Force copy of input img (#297)

fa470ebc · Miguel Mendez · GitHub · 02f2573a · fa470ebc
Unverified Commit fa470ebc authored May 24, 2020 by Miguel Mendez Committed by GitHub May 24, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/image/photometric.py mmcv/image/photometric.py +1 -1

No files found.
--- a/mmcv/image/photometric.py
+++ b/mmcv/image/photometric.py
@@ -14,7 +14,7 @@ def imnormalize(img, mean, std, to_rgb=True):
    Returns:
        ndarray: The normalized image.
    """
-    img = np.float32(img) if img.dtype != np.float32 else img.copy()
+    img = img.copy().astype(np.float32)
    return imnormalize_(img, mean, std, to_rgb)