From 9a87bdbae194d2aacc65834202bc5554f73694f0 Mon Sep 17 00:00:00 2001
From: sachalin <sacha@lin.com>
Date: Wed, 23 Nov 2022 14:04:20 +0100
Subject: [PATCH] linear interpolation recasting
---
latent_blending.py | 20 +++++++++++++++++---
1 file changed, 17 insertions(+), 3 deletions(-)
diff --git a/latent_blending.py b/latent_blending.py
index 4c24a9c..cd3dad2 100644
--- a/latent_blending.py
+++ b/latent_blending.py
@@ -832,16 +832,30 @@ def interpolate_linear(p0, p1, fract_mixing):
Helper function to mix two variables using standard linear interpolation.
Args:
p0:
- First tensor for interpolation
+ First tensor / np.ndarray for interpolation
p1:
- Second tensor for interpolation
+ Second tensor / np.ndarray for interpolation
fract_mixing: float
Mixing coefficient of interval [0, 1].
0 will return in p0
1 will return in p1
0.x will return a linear mix between both.
"""
- return (1-fract_mixing) * p0 + fract_mixing * p1
+ reconvert_uint8 = False
+ if type(p0) is np.ndarray and p0.dtype == 'uint8':
+ reconvert_uint8 = True
+ p0 = p0.astype(np.float64)
+
+ if type(p1) is np.ndarray and p1.dtype == 'uint8':
+ reconvert_uint8 = True
+ p1 = p1.astype(np.float64)
+
+ interp = (1-fract_mixing) * p0 + fract_mixing * p1
+
+ if reconvert_uint8:
+ interp = np.clip(interp, 0, 255).astype(np.uint8)
+
+ return interp
def add_frames_linear_interp(