(self)
| 191 | return model |
| 192 | |
| 193 | def load_models(self): |
| 194 | model_path = get_model_dir("liveportrait") |
| 195 | if not os.path.exists(model_path): |
| 196 | os.mkdir(model_path) |
| 197 | |
| 198 | model_config_path = os.path.join(current_directory, 'LivePortrait', 'config', 'models.yaml') |
| 199 | model_config = yaml.safe_load(open(model_config_path, 'r')) |
| 200 | |
| 201 | appearance_feature_extractor = self.load_model(model_config, 'appearance_feature_extractor') |
| 202 | motion_extractor = self.load_model(model_config, 'motion_extractor') |
| 203 | warping_module = self.load_model(model_config, 'warping_module') |
| 204 | spade_generator = self.load_model(model_config, 'spade_generator') |
| 205 | stitching_retargeting_module = self.load_model(model_config, 'stitching_retargeting_module') |
| 206 | |
| 207 | self.pipeline = LivePortraitWrapper(InferenceConfig(), appearance_feature_extractor, motion_extractor, warping_module, spade_generator, stitching_retargeting_module) |
| 208 | |
| 209 | def get_detect_model(self): |
| 210 | if self.detect_model == None: |
no test coverage detected