jhj0517
commited on
Commit
·
90303f6
1
Parent(s):
de78657
Remove meaningless params and edit function
Browse files
modules/live_portrait/live_portrait_inferencer.py
CHANGED
@@ -246,28 +246,21 @@ class LivePortraitInferencer:
|
|
246 |
tracking_src_vid: bool,
|
247 |
animate_without_vid: bool,
|
248 |
crop_factor: float,
|
249 |
-
|
250 |
driving_images: Optional[List[np.ndarray]] = None,
|
251 |
-
motion_link: Optional[List] = None,
|
252 |
progress: gr.Progress = gr.Progress()
|
253 |
):
|
254 |
src_length = 1
|
255 |
|
256 |
-
if
|
257 |
-
|
258 |
-
|
259 |
-
else:
|
260 |
-
return None, None
|
261 |
-
|
262 |
-
if src_images is not None:
|
263 |
-
src_length = len(src_images)
|
264 |
-
if id(src_images) != id(self.src_images) or self.crop_factor != crop_factor:
|
265 |
self.crop_factor = crop_factor
|
266 |
-
self.
|
267 |
if 1 < src_length:
|
268 |
-
self.psi_list = self.prepare_source(
|
269 |
else:
|
270 |
-
self.psi_list = [self.prepare_source(
|
271 |
|
272 |
driving_length = 0
|
273 |
if driving_images is not None:
|
|
|
246 |
tracking_src_vid: bool,
|
247 |
animate_without_vid: bool,
|
248 |
crop_factor: float,
|
249 |
+
src_image_list: Optional[List[np.ndarray]] = None,
|
250 |
driving_images: Optional[List[np.ndarray]] = None,
|
|
|
251 |
progress: gr.Progress = gr.Progress()
|
252 |
):
|
253 |
src_length = 1
|
254 |
|
255 |
+
if src_image_list is not None:
|
256 |
+
src_length = len(src_image_list)
|
257 |
+
if id(src_image_list) != id(self.src_image_list) or self.crop_factor != crop_factor:
|
|
|
|
|
|
|
|
|
|
|
|
|
258 |
self.crop_factor = crop_factor
|
259 |
+
self.src_image_list = src_image_list
|
260 |
if 1 < src_length:
|
261 |
+
self.psi_list = [self.prepare_source(src, crop_factor, True, tracking_src_vid) for src in src_image_list]
|
262 |
else:
|
263 |
+
self.psi_list = [self.prepare_source(src, crop_factor) for src in src_image_list]
|
264 |
|
265 |
driving_length = 0
|
266 |
if driving_images is not None:
|