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:
|