Revert "a hack to for onnxruntime to use cuda"
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user