(self, save_mode, face_model_name, select_face_index, image=None, face_model=None, det_size=(640, 640))
| 629 | CATEGORY = "🌌 ReActor" |
| 630 | |
| 631 | def save_model(self, save_mode, face_model_name, select_face_index, image=None, face_model=None, det_size=(640, 640)): |
| 632 | if save_mode and image is not None: |
| 633 | source = tensor_to_pil(image) |
| 634 | source = cv2.cvtColor(np.array(source), cv2.COLOR_RGB2BGR) |
| 635 | apply_patch(1) |
| 636 | logger.status("Building Face Model...") |
| 637 | face_model_raw = analyze_faces(source, det_size) |
| 638 | if len(face_model_raw) == 0: |
| 639 | det_size_half = half_det_size(det_size) |
| 640 | face_model_raw = analyze_faces(source, det_size_half) |
| 641 | try: |
| 642 | face_model = face_model_raw[select_face_index] |
| 643 | except: |
| 644 | logger.error("No face(s) found") |
| 645 | return face_model_name |
| 646 | logger.status("--Done!--") |
| 647 | if save_mode and (face_model != "none" or face_model is not None): |
| 648 | face_model_path = os.path.join(self.output_dir, face_model_name + ".safetensors") |
| 649 | save_face_model(face_model,face_model_path) |
| 650 | if image is None and face_model is None: |
| 651 | logger.error("Please provide `face_model` or `image`") |
| 652 | return face_model_name |
| 653 | |
| 654 | |
| 655 | class RestoreFace: |
nothing calls this directly
no test coverage detected