jhj0517
commited on
Commit
·
de78657
1
Parent(s):
dc1090e
Add type hint and remove meaningless params
Browse files
modules/live_portrait/live_portrait_inferencer.py
CHANGED
|
@@ -10,7 +10,7 @@ import gradio as gr
|
|
| 10 |
from gradio_i18n import Translate, gettext as _
|
| 11 |
from ultralytics.utils import LOGGER as ultralytics_logger
|
| 12 |
from enum import Enum
|
| 13 |
-
from typing import Union
|
| 14 |
|
| 15 |
from modules.utils.paths import *
|
| 16 |
from modules.utils.image_helper import *
|
|
@@ -241,19 +241,16 @@ class LivePortraitInferencer:
|
|
| 241 |
raise
|
| 242 |
|
| 243 |
def create_video(self,
|
| 244 |
-
retargeting_eyes,
|
| 245 |
-
retargeting_mouth,
|
| 246 |
-
|
| 247 |
-
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
|
| 251 |
-
|
| 252 |
-
|
| 253 |
-
|
| 254 |
-
progress=gr.Progress()):
|
| 255 |
-
if not turn_on:
|
| 256 |
-
return None, None
|
| 257 |
src_length = 1
|
| 258 |
|
| 259 |
if src_images is None:
|
|
@@ -272,11 +269,6 @@ class LivePortraitInferencer:
|
|
| 272 |
else:
|
| 273 |
self.psi_list = [self.prepare_source(src_images, crop_factor)]
|
| 274 |
|
| 275 |
-
cmd_list, cmd_length = self.parsing_command(command, motion_link)
|
| 276 |
-
if cmd_list is None:
|
| 277 |
-
return None,None
|
| 278 |
-
cmd_idx = 0
|
| 279 |
-
|
| 280 |
driving_length = 0
|
| 281 |
if driving_images is not None:
|
| 282 |
if id(driving_images) != id(self.driving_images):
|
|
@@ -287,7 +279,7 @@ class LivePortraitInferencer:
|
|
| 287 |
total_length = max(driving_length, src_length)
|
| 288 |
|
| 289 |
if animate_without_vid:
|
| 290 |
-
total_length =
|
| 291 |
|
| 292 |
c_i_es = ExpressionSet()
|
| 293 |
c_o_es = ExpressionSet()
|
|
|
|
| 10 |
from gradio_i18n import Translate, gettext as _
|
| 11 |
from ultralytics.utils import LOGGER as ultralytics_logger
|
| 12 |
from enum import Enum
|
| 13 |
+
from typing import Union, List, Dict, Tuple
|
| 14 |
|
| 15 |
from modules.utils.paths import *
|
| 16 |
from modules.utils.image_helper import *
|
|
|
|
| 241 |
raise
|
| 242 |
|
| 243 |
def create_video(self,
|
| 244 |
+
retargeting_eyes: bool,
|
| 245 |
+
retargeting_mouth: bool,
|
| 246 |
+
tracking_src_vid: bool,
|
| 247 |
+
animate_without_vid: bool,
|
| 248 |
+
crop_factor: float,
|
| 249 |
+
src_images: Optional[List[np.ndarray]] = None,
|
| 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 src_images is None:
|
|
|
|
| 269 |
else:
|
| 270 |
self.psi_list = [self.prepare_source(src_images, crop_factor)]
|
| 271 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 272 |
driving_length = 0
|
| 273 |
if driving_images is not None:
|
| 274 |
if id(driving_images) != id(self.driving_images):
|
|
|
|
| 279 |
total_length = max(driving_length, src_length)
|
| 280 |
|
| 281 |
if animate_without_vid:
|
| 282 |
+
total_length = total_length
|
| 283 |
|
| 284 |
c_i_es = ExpressionSet()
|
| 285 |
c_o_es = ExpressionSet()
|