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 1/2] 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 From 4d3bdf8ee197d8bda47559938d29d1811e994af3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Phan=20Tu=E1=BA=A5n=20Anh?= Date: Mon, 26 Jun 2023 14:28:38 +0200 Subject: [PATCH 2/2] comments to understand why torch --- recognition/arcface_onnx.py | 2 +- recognition/main.py | 2 +- recognition/scrfd.py | 2 +- refacer.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/recognition/arcface_onnx.py b/recognition/arcface_onnx.py index e77b421..5cda3ec 100644 --- a/recognition/arcface_onnx.py +++ b/recognition/arcface_onnx.py @@ -6,7 +6,7 @@ import numpy as np import cv2 -import torch +import torch # hack to use onnxruntime with cuda import onnx import onnxruntime import face_align diff --git a/recognition/main.py b/recognition/main.py index b353f68..bddf777 100644 --- a/recognition/main.py +++ b/recognition/main.py @@ -5,7 +5,7 @@ import os.path as osp import argparse import cv2 import numpy as np -import torch +import torch # hack to use onnxruntime with cuda import onnxruntime from scrfd import SCRFD from arcface_onnx import ArcFaceONNX diff --git a/recognition/scrfd.py b/recognition/scrfd.py index 2cad754..2a1a156 100644 --- a/recognition/scrfd.py +++ b/recognition/scrfd.py @@ -2,7 +2,7 @@ from __future__ import division import datetime import numpy as np -import torch +import torch # hack to use onnxruntime with cuda import onnxruntime import os import os.path as osp diff --git a/refacer.py b/refacer.py index 409c378..97795cb 100644 --- a/refacer.py +++ b/refacer.py @@ -1,5 +1,5 @@ import cv2 -import torch +import torch # hack to use onnxruntime with cuda import onnxruntime as rt import sys from insightface.app import FaceAnalysis