Merge pull request modelscope#167 from ultimatech-cn/patch-1

Update preprocessing.py
0x35 · Sep 2, 2023 · 01143b2 · 01143b2
2 parents 5870b12 + 72018ed
commit 01143b2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/facechain/data_process/preprocessing.py b/facechain/data_process/preprocessing.py
@@ -349,8 +349,10 @@ def __call__(self, imdir):
         # copy to the ensembledir for reading
         for i, index in enumerate(ref_indexes[:4]):
             save_path = os.path.join(ensembledir, f"best_roop_image_{str(i)}.jpg")
-            os.system(f"cp -rf {os.path.join(imdir, selected_paths[index])} {save_path}")
-
+            #os.system(f"cp -rf {os.path.join(imdir, selected_paths[index])} {save_path}")
+            if os.path.exists(save_path):
+                os.remove(save_path)
+            shutil.copy2(os.path.join(imdir, selected_paths[index]), save_path)
         cnt = 0
         tmp_path = os.path.join(savedir, 'tmp.png')
         for imname in _selected_paths: