Ver Fonte

Script for image face swapping(You can choose to add or not add a watermark )

NNNNAI há 4 anos atrás
pai
commit
e7496dad94
2 ficheiros alterados com 3 adições e 2 exclusões
  1. 2 1
      test_wholeimage_swapmutil.py
  2. 1 1
      test_wholeimage_swapsingle.py

+ 2 - 1
test_wholeimage_swapmutil.py

@@ -73,7 +73,8 @@ if __name__ == '__main__':
         swap_result = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
 
-    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmutil.jpg'))
+
+    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmutil.jpg'),opt.no_simswaplogo)
     print(' ')
 
     print('************ Done ! ************')

+ 1 - 1
test_wholeimage_swapsingle.py

@@ -72,7 +72,7 @@ if __name__ == '__main__':
         swap_result = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
 
-    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapsingle.jpg'))
+    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapsingle.jpg'),opt.no_simswaplogo)
 
     print(' ')