Merge pull request #55 from phineas-pta/test-torch

a hack to for onnxruntime to use cuda

recognition/arcface_onnx.py

@@ -6,6 +6,7 @@
 
 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,6 +5,7 @@ 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 onnx
+import torch # hack to use onnxruntime with cuda
 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