change scipy.misc.imsave to imageio.imwrite

2020-06-17 22:15:39 +01:00 · 2020-06-17 22:15:39 +01:00 · accb3e2146
parent b2e956cabd
commit accb3e2146
1 changed files with 2 additions and 2 deletions
--- a/torchreid/data/datasets/image/cuhk03.py
+++ b/torchreid/data/datasets/image/cuhk03.py
@ -103,7 +103,7 @@ class CUHK03(ImageDataset):
            return

        import h5py
-        from scipy.misc import imsave
+        import imageio
        from scipy.io import loadmat

        mkdir_if_missing(self.imgs_detected_dir)
@ -136,7 +136,7 @@ class CUHK03(ImageDataset):
                )
                img_path = osp.join(save_dir, img_name)
                if not osp.isfile(img_path):
-                    imsave(img_path, img)
+                    imageio.imwrite(img_path, img)
                img_paths.append(img_path)
            return img_paths