Преглед изворни кода

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

NNNNAI пре 4 година
родитељ
комит
e7496dad94
2 измењених фајлова са 3 додато и 2 уклоњено
  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 = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
         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(' ')
 
 
     print('************ Done ! ************')
     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 = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
         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(' ')
     print(' ')