()
| 9 | |
| 10 | |
| 11 | def load_models(): |
| 12 | try: |
| 13 | _tf.config.face_detector_model = _get_model_path("scrfd_2.5g.onnx") |
| 14 | _tf.config.face_embedder_model = _get_model_path("arcface_w600k_r50.onnx") |
| 15 | _tf.config.face_swapper_model = _get_model_path("inswapper_128_fp16.onnx") |
| 16 | _tf.config.face_enhancer_model = _get_model_path("gfpgan_1.4.onnx") |
| 17 | _tf.prepare() |
| 18 | return True |
| 19 | except BaseException as _: |
| 20 | return False |
| 21 | |
| 22 | |
| 23 | @lru_cache(maxsize=12) |
no test coverage detected