Revert "a hack to for onnxruntime to use cuda"

2023-06-27 20:39:24 +02:00
parent 747726fc02
commit e7907c45b6
4 changed files with 2 additions and 4 deletions
--- 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