CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

color transfer sot : optimization

Browse Source
This commit is contained in:
Colombo
2020-04-15 18:13:03 +04:00
parent f8580928ed
commit 6bbc607312
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/imagelib/color_transfer.py
+2 -1
View File
@@ -35,8 +35,9 @@ def color_transfer_sot(src,trg, steps=10, batch_size=5, reg_sigmaXY=16.0, reg_si
h,w,c = src.shape h,w,c = src.shape
new_src = src.copy() new_src = src.copy()
advect = np.empty ( (h*w,c), dtype=src_dtype )
for step in range (steps): for step in range (steps):
advect = np.zeros ( (h*w,c), dtype=src_dtype ) advect.fill(0)
for batch in range (batch_size): for batch in range (batch_size):
dir = np.random.normal(size=c).astype(src_dtype) dir = np.random.normal(size=c).astype(src_dtype)
dir /= npla.norm(dir) dir /= npla.norm(dir)