diff --git a/recognition/arcface_onnx.py b/recognition/arcface_onnx.py index 5cda3ec..4b563e5 100644 --- a/recognition/arcface_onnx.py +++ b/recognition/arcface_onnx.py @@ -6,7 +6,6 @@ import numpy as np import cv2 -import torch # hack to use onnxruntime with cuda import onnx import onnxruntime import face_align diff --git a/recognition/main.py b/recognition/main.py index bddf777..9ba74ae 100644 --- a/recognition/main.py +++ b/recognition/main.py @@ -5,7 +5,6 @@ import os.path as osp import argparse import cv2 import numpy as np -import torch # hack to use onnxruntime with cuda import onnxruntime from scrfd import SCRFD from arcface_onnx import ArcFaceONNX diff --git a/recognition/scrfd.py b/recognition/scrfd.py index 2a1a156..0b79556 100644 --- a/recognition/scrfd.py +++ b/recognition/scrfd.py @@ -2,7 +2,7 @@ from __future__ import division import datetime import numpy as np -import torch # hack to use onnxruntime with cuda +#import onnx import onnxruntime import os import os.path as osp diff --git a/refacer.py b/refacer.py index 97795cb..5be6d70 100644 --- a/refacer.py +++ b/refacer.py @@ -1,11 +1,11 @@ import cv2 -import torch # hack to use onnxruntime with cuda import onnxruntime as rt import sys from insightface.app import FaceAnalysis sys.path.insert(1, './recognition') from scrfd import SCRFD from arcface_onnx import ArcFaceONNX +import os.path as osp import os from pathlib import Path from tqdm import tqdm