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/scrfd.py
+++ b/recognition/scrfd.py
@@ -2,7 +2,7 @@
 from __future__ import division
 import datetime
 import numpy as np
-#import onnx
+import torch
 import onnxruntime
 import os
 import os.path as osp