admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Incredible speed boost with TensorrtExecutionProvider #5.

Browse Source
This commit is contained in:
Xavi Vinaixa
2023-06-06 08:10:51 +02:00
parent 4a24df4755
commit a8d2941293
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
refacer.py
View File

@@ -20,7 +20,7 @@ from insightface.app.common import Face
from insightface.utils.storage import ensure_available from insightface.utils.storage import ensure_available
class RefacerMode(Enum): class RefacerMode(Enum):
CPU, CUDA, COREML, TENSORRT = range(1, 4) CPU, CUDA, COREML, TENSORRT = range(1, 5)
class Refacer: class Refacer:
def __init__(self,force_cpu=False): def __init__(self,force_cpu=False):