Delete refacer.py

refacer.py

@@ -1,77 +0,0 @@
-import cv2
-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
-import ffmpeg
-import random
-import multiprocessing as mp
-from concurrent.futures import ThreadPoolExecutor
-from insightface.model_zoo.inswapper import INSwapper
-import psutil
-from enum import Enum
-from insightface.app.common import Face
-from insightface.utils.storage import ensure_available
-import re
-import subprocess
-
-class RefacerMode(Enum):
-     CPU, CUDA, COREML, TENSORRT = range(1, 5)
-
-class Refacer:
-    def __init__(self,force_cpu=False,colab_performance=False):
-        self.first_face = False
-        self.force_cpu = force_cpu
-        self.colab_performance = colab_performance
-        self.__check_encoders()
-        self.__check_providers()
-        self.total_mem = psutil.virtual_memory().total
-        self.__init_apps()
-
-    def __check_providers(self):
-        if self.force_cpu :
-            self.providers = ['CPUExecutionProvider']
-        else:
-            self.providers = rt.get_available_providers()
-        rt.set_default_logger_severity(4)
-        self.sess_options = rt.SessionOptions()
-        self.sess_options.execution_mode = rt.ExecutionMode.ORT_SEQUENTIAL
-        self.sess_options.graph_optimization_level = rt.GraphOptimizationLevel.ORT_ENABLE_ALL
-
-        if len(self.providers) == 1 and 'CPUExecutionProvider' in self.providers:
-            self.mode = RefacerMode.CPU
-            self.use_num_cpus = mp.cpu_count()-1
-            self.sess_options.intra_op_num_threads = int(self.use_num_cpus/3)
-            print(f"CPU mode with providers {self.providers}")
-        elif self.colab_performance:
-            self.mode = RefacerMode.TENSORRT
-            self.use_num_cpus = mp.cpu_count()-1
-            self.sess_options.intra_op_num_threads = int(self.use_num_cpus/3)
-            print(f"Colab Mode (TENSORRT): {self.providers}")
-        else:
-            self.mode = RefacerMode.CUDA
-            self.use_num_cpus = mp.cpu_count()-2
-            self.sess_options.intra_op_num_threads = int(self.use_num_cpus/3)
-            print(f"CUDA Mode with providers {self.providers}")
-
-    def __init_apps(self):
-        self.face_app = FaceAnalysis()
-        self.face_app.prepare(ctx_id=0, det_size=(640, 640))
-
-        self.inswapper = INSwapper()
-
-    def reface(self, video_path, faces):
-        self.first_face = True
-        input_video = cv2.VideoCapture(video_path)
-        video_fps = input_video.get(cv2.CAP_PROP_FPS)
-        width = int(input_video.get(cv2.CAP_PROP_FRAME_WIDTH))
-        height = int(input_video.get(cv2.CAP_PROP_FRAME_HEIGHT))
-        output_video = cv2.VideoWriter('refaced_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'), video_fps, (width, height))
-
-        return output_video