From e7907c45b6e4862e941fcd7dac09e475330f520c Mon Sep 17 00:00:00 2001 From: xaviviro Date: Tue, 27 Jun 2023 20:39:24 +0200 Subject: [PATCH] Revert "a hack to for onnxruntime to use cuda" --- recognition/arcface_onnx.py | 1 - recognition/main.py | 1 - recognition/scrfd.py | 2 +- refacer.py | 2 +- 4 files changed, 2 insertions(+), 4 deletions(-) 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