From 91144be667ff11b938738f58331dbcae7ea7c6a2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Phan=20Tu=E1=BA=A5n=20Anh?= Date: Mon, 26 Jun 2023 14:19:44 +0200 Subject: [PATCH] import torch --- recognition/arcface_onnx.py | 1 + recognition/main.py | 1 + recognition/scrfd.py | 2 +- refacer.py | 2 +- 4 files changed, 4 insertions(+), 2 deletions(-) diff --git a/recognition/arcface_onnx.py b/recognition/arcface_onnx.py index 4b563e5..e77b421 100644 --- 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 diff --git a/recognition/main.py b/recognition/main.py index 9ba74ae..b353f68 100644 --- 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 diff --git a/recognition/scrfd.py b/recognition/scrfd.py index 0b79556..2cad754 100644 --- 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 diff --git a/refacer.py b/refacer.py index 5be6d70..409c378 100644 --- a/refacer.py +++ b/refacer.py @@ -1,11 +1,11 @@ import cv2 +import torch 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