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