import torch

2023-06-26 14:19:44 +02:00
parent 789cf77081
commit 91144be667
4 changed files with 4 additions and 2 deletions
--- a/recognition/main.py
+++ b/recognition/main.py
@@ -5,6 +5,7 @@ import os.path as osp
 import argparse
 import cv2
 import numpy as np
+import torch
 import onnxruntime
 from scrfd import SCRFD
 from arcface_onnx import ArcFaceONNX