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/arcface_onnx.py
+++ b/recognition/arcface_onnx.py
@@ -6,6 +6,7 @@

 import numpy as np
 import cv2
+import torch
 import onnx
 import onnxruntime
 import face_align