diff --git a/refacer.py b/refacer.py index 1f5d161..6307237 100644 --- a/refacer.py +++ b/refacer.py @@ -57,6 +57,7 @@ class Refacer: self.use_num_cpus = 1 self.sess_options.intra_op_num_threads = 1 print(f"CUDA mode with providers {self.providers}") + """ elif 'TensorrtExecutionProvider' in self.providers: self.mode = RefacerMode.TENSORRT #self.use_num_cpus = 1 @@ -64,6 +65,7 @@ class Refacer: self.use_num_cpus = mp.cpu_count()-1 self.sess_options.intra_op_num_threads = int(self.use_num_cpus/3) print(f"TENSORRT mode with providers {self.providers}") + """ def __init_apps(self): @@ -193,7 +195,7 @@ class Refacer: VIDEO_CODECS = [ #'h264_videotoolbox', #osx HW acceleration - 'h264_nvenc', #NVIDIA HW acceleration + #'h264_nvenc', #NVIDIA HW acceleration #'h264_qsv', #Intel HW acceleration #'h264_vaapi', #Intel HW acceleration #'h264_omx', #HW acceleration diff --git a/testsrc.mp4 b/testsrc.mp4 new file mode 100644 index 0000000..d77c980 Binary files /dev/null and b/testsrc.mp4 differ