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
- 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:
257
- if motion_link is not None:
258
- self.psi_list = [motion_link[0]]
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.src_images = src_images
267
  if 1 < src_length:
268
- self.psi_list = self.prepare_source(src_images, crop_factor, True, tracking_src_vid)
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:
 
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: