From 6bbc60731289d02107af51fba5b7c55ccc8e8085 Mon Sep 17 00:00:00 2001 From: Colombo Date: Wed, 15 Apr 2020 18:13:03 +0400 Subject: [PATCH] color transfer sot : optimization --- core/imagelib/color_transfer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/imagelib/color_transfer.py b/core/imagelib/color_transfer.py index 22f6876..a329f7a 100644 --- a/core/imagelib/color_transfer.py +++ b/core/imagelib/color_transfer.py @@ -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 new_src = src.copy() + advect = np.empty ( (h*w,c), dtype=src_dtype ) for step in range (steps): - advect = np.zeros ( (h*w,c), dtype=src_dtype ) + advect.fill(0) for batch in range (batch_size): dir = np.random.normal(size=c).astype(src_dtype) dir /= npla.norm(dir)