using mkstemp

samkit-jain · Apr 30, 2019 · 1e6b399 · 1e6b399
1 parent accd457
commit 1e6b399
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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.mktemp(dir=label_folder, suffix='.png')
+                image_dest = tempfile.mkstemp(dir=label_folder, suffix='.png')
 
                 cv2.imwrite(f'{image_dest}', image.T)
 

diff --git a/Python/drawer.py b/Python/drawer.py
@@ -47,7 +47,7 @@ def get_contours(self):
         main_image = cv2.GaussianBlur(src=main_image, ksize=(5, 5), sigmaX=0)
 
         # threshold the image
-        ret, main_image = cv2.threshold(src=main_image, thresh=127, maxval=255, type=cv2.THRESH_BINARY)
+        _, main_image = cv2.threshold(src=main_image, thresh=127, maxval=255, type=cv2.THRESH_BINARY)
 
         # find contours in the image
         contours, _ = cv2.findContours(image=main_image.copy(), mode=cv2.RETR_EXTERNAL, method=cv2.CHAIN_APPROX_SIMPLE)