Commit @6d3bc384c91701c7b1448a43c6ca1b9b73bfb23e

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-06-29

something is weird about this gamma correction

@6d3bc384c91701c7b1448a43c6ca1b9b73bfb23e

84f45a0

6d3bc38

model/Generator.py

--- model/Generator.py

+++ model/Generator.py


         return ret
 
     def binary_diff_mask(self, clean, dirty, thresold=0.1):
-        clean = torch.pow(clean, 1/2.2)
-        dirty = torch.pow(dirty, 1/2.2)
+        clean = torch.pow(clean, 2.2)
+        dirty = torch.pow(dirty, 2.2)
         diff = torch.abs(clean - dirty)
         diff = torch.sum(diff, dim=1)
         # this line is certainly cause problem for quantization

Add a comment

Open 0
Closed 0

List

...	...	@@ -35,8 +35,8 @@
35	35	return ret
36	36
37	37	def binary_diff_mask(self, clean, dirty, thresold=0.1):
38		- clean = torch.pow(clean, 1/2.2)
39		- dirty = torch.pow(dirty, 1/2.2)
	38	+ clean = torch.pow(clean, 2.2)
	39	+ dirty = torch.pow(dirty, 2.2)
40	40	diff = torch.abs(clean - dirty)
41	41	diff = torch.sum(diff, dim=1)
42	42	# this line is certainly cause problem for quantization

Delete comment