Merge pull request #59 from xaviviro/revert-55-test-torch

Revert "a hack to for onnxruntime to use cuda"

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

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

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

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