Commit @f4b369127b142b5a1e4c85ebf6f6991f787b3362

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-07-12

data cleaning

@f4b369127b142b5a1e4c85ebf6f6991f787b3362

4346ed6

f4b3691

binary mask map test.py

--- binary mask map test.py

+++ binary mask map test.py


 def process(i, j):
     print(i)
     clean_img_copy = shift_img(clean_img, (i-k)/4, (j-k)/4)
-    binary_diff_mask_img = binary_diff_mask(dirty_img / 255, clean_img_copy / 255, threshold=0.2)
-    cv2.imwrite(f"test/test_img{(i-k)/4}-{(j-k)/4}.png", binary_diff_mask_img*255)
+    binary_diff_mask_img = binary_diff_mask(dirty_img / 255, clean_img_copy / 255, threshold=0.3)
+    success = cv2.imwrite(f"test/test_img_x{(i-k)/4}-y{(j-k)/4}.png", binary_diff_mask_img*255)
+    if not success:
+        print(f"Failed to save image at test/test_img{(i - k) / 4}-{(j - k) / 4}.png")
+
 
 with concurrent.futures.ProcessPoolExecutor() as executor:
     for i in range(k*2):

Add a comment

Open 0
Closed 0

List

...	...	@@ -40,8 +40,11 @@
40	40	def process(i, j):
41	41	print(i)
42	42	clean_img_copy = shift_img(clean_img, (i-k)/4, (j-k)/4)
43		- binary_diff_mask_img = binary_diff_mask(dirty_img / 255, clean_img_copy / 255, threshold=0.2)
44		- cv2.imwrite(f"test/test_img{(i-k)/4}-{(j-k)/4}.png", binary_diff_mask_img*255)
	43	+ binary_diff_mask_img = binary_diff_mask(dirty_img / 255, clean_img_copy / 255, threshold=0.3)
	44	+ success = cv2.imwrite(f"test/test_img_x{(i-k)/4}-y{(j-k)/4}.png", binary_diff_mask_img*255)
	45	+ if not success:
	46	+ print(f"Failed to save image at test/test_img{(i - k) / 4}-{(j - k) / 4}.png")
	47	+
45	48
46	49	with concurrent.futures.ProcessPoolExecutor() as executor:
47	50	for i in range(k*2):

Delete comment