| 6 | |
| 7 | |
| 8 | class FaceModel(nn.Module): |
| 9 | def __init__(self): |
| 10 | super(FaceModel, self).__init__() |
| 11 | self.app = FaceAnalysis( |
| 12 | name='antelopev2', root='.', providers=['CUDAExecutionProvider', 'CPUExecutionProvider', ] |
| 13 | ) |
| 14 | self.app.prepare(ctx_id=-1, det_size=(640, 640)) |
| 15 | self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx') |
| 16 | self.handler_ante.prepare(ctx_id=-1) |
| 17 | |
| 18 | self.face_helper = FaceRestoreHelper( |
| 19 | upscale_factor=1, |
| 20 | face_size=512, |
| 21 | crop_ratio=(1, 1), |
| 22 | det_model='retinaface_resnet50', |
| 23 | save_ext='png', |
| 24 | device="cpu", |
| 25 | ) |
| 26 | self.face_helper.face_parse = init_parsing_model(model_name='bisenet', device="cpu") |
no outgoing calls
no test coverage detected