Revert "a hack to for onnxruntime to use cuda"

This commit is contained in:
xaviviro
2023-06-27 20:39:24 +02:00
committed by GitHub
parent 747726fc02
commit e7907c45b6
4 changed files with 2 additions and 4 deletions

View File

@@ -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