bug fix

samkit-jain · Apr 30, 2019 · a7492b9 · a7492b9
1 parent 1e6b399
commit a7492b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Python/dataset.py b/Python/dataset.py
@@ -168,7 +168,7 @@ def idx_to_image(self, image_file: str, label_file: str, label_mapping: Dict[str
             # save images to data directory
             for label, image in zip(labels, images):
                 label_folder = osp.abspath(osp.join(self.data_dir_path, label))
-                image_dest = tempfile.mkstemp(dir=label_folder, suffix='.png')
+                image_dest = tempfile.mkstemp(dir=label_folder, suffix='.png')[1]
 
                 cv2.imwrite(f'{image_dest}', image.T)