diff --git a/test_video_swap_mutilspecific.py b/test_video_swap_multispecific.py similarity index 92% rename from test_video_swap_mutilspecific.py rename to test_video_swap_multispecific.py index 0b58456..5426ca8 100644 --- a/test_video_swap_mutilspecific.py +++ b/test_video_swap_multispecific.py @@ -8,8 +8,8 @@ import torch.nn.functional as F from torchvision import transforms from models.models import create_model from options.test_options import TestOptions -from insightface_func.face_detect_crop_mutil import Face_detect_crop -from util.videoswap_mutilspecific import video_swap +from insightface_func.face_detect_crop_multi import Face_detect_crop +from util.videoswap_multispecific import video_swap import os import glob @@ -37,7 +37,7 @@ if __name__ == '__main__': start_epoch, epoch_iter = 1, 0 crop_size = 224 - mutilsepcific_dir = opt.mutilsepcific_dir + multisepcific_dir = opt.multisepcific_dir torch.nn.Module.dump_patches = True model = create_model(opt) model.eval() @@ -49,7 +49,7 @@ if __name__ == '__main__': # The specific person to be swapped(source) source_specific_id_nonorm_list = [] - source_path = os.path.join(mutilsepcific_dir,'source','*') + source_path = os.path.join(multisepcific_dir,'SRC_*') source_specific_images_path = sorted(glob.glob(source_path)) for source_specific_image_path in source_specific_images_path: @@ -68,7 +68,7 @@ if __name__ == '__main__': # The person who provides id information (list) target_id_norm_list = [] - target_path = os.path.join(mutilsepcific_dir,'target','*') + target_path = os.path.join(multisepcific_dir,'DST_*') target_images_path = sorted(glob.glob(target_path)) for target_image_path in target_images_path: diff --git a/test_video_swapmutil.py b/test_video_swapmulti.py similarity index 97% rename from test_video_swapmutil.py rename to test_video_swapmulti.py index de09bfc..14a98ee 100644 --- a/test_video_swapmutil.py +++ b/test_video_swapmulti.py @@ -8,7 +8,7 @@ import torch.nn.functional as F from torchvision import transforms from models.models import create_model from options.test_options import TestOptions -from insightface_func.face_detect_crop_mutil import Face_detect_crop +from insightface_func.face_detect_crop_multi import Face_detect_crop from util.videoswap import video_swap import os diff --git a/test_video_swapspecific.py b/test_video_swapspecific.py index e12ecb9..dbcc4e0 100644 --- a/test_video_swapspecific.py +++ b/test_video_swapspecific.py @@ -8,7 +8,7 @@ import torch.nn.functional as F from torchvision import transforms from models.models import create_model from options.test_options import TestOptions -from insightface_func.face_detect_crop_mutil import Face_detect_crop +from insightface_func.face_detect_crop_multi import Face_detect_crop from util.videoswap_specific import video_swap import os diff --git a/test_wholeimage_swap_mutilspecific.py b/test_wholeimage_swap_multispecific.py similarity index 94% rename from test_wholeimage_swap_mutilspecific.py rename to test_wholeimage_swap_multispecific.py index e9a243a..a5c6622 100644 --- a/test_wholeimage_swap_mutilspecific.py +++ b/test_wholeimage_swap_multispecific.py @@ -8,7 +8,7 @@ import torch.nn.functional as F from torchvision import transforms from models.models import create_model from options.test_options import TestOptions -from insightface_func.face_detect_crop_mutil import Face_detect_crop +from insightface_func.face_detect_crop_multi import Face_detect_crop from util.reverse2original import reverse2wholeimage import os from util.add_watermark import watermark_image @@ -39,7 +39,7 @@ if __name__ == '__main__': start_epoch, epoch_iter = 1, 0 crop_size = 224 - mutilsepcific_dir = opt.mutilsepcific_dir + multisepcific_dir = opt.multisepcific_dir torch.nn.Module.dump_patches = True logoclass = watermark_image('./simswaplogo/simswaplogo.png') @@ -56,7 +56,7 @@ if __name__ == '__main__': # The specific person to be swapped(source) source_specific_id_nonorm_list = [] - source_path = os.path.join(mutilsepcific_dir,'source','*') + source_path = os.path.join(multisepcific_dir,'SRC_*') source_specific_images_path = sorted(glob.glob(source_path)) for source_specific_image_path in source_specific_images_path: @@ -75,7 +75,7 @@ if __name__ == '__main__': # The person who provides id information (list) target_id_norm_list = [] - target_path = os.path.join(mutilsepcific_dir,'target','*') + target_path = os.path.join(multisepcific_dir,'DST_*') target_images_path = sorted(glob.glob(target_path)) for target_image_path in target_images_path: @@ -135,7 +135,7 @@ if __name__ == '__main__': if len(swap_result_list) !=0: - reverse2wholeimage(swap_result_list, swap_result_matrix_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swap_mutilspecific.jpg'), opt.no_simswaplogo) + reverse2wholeimage(swap_result_list, swap_result_matrix_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swap_multispecific.jpg'), opt.no_simswaplogo) print(' ') diff --git a/test_wholeimage_swapmutil.py b/test_wholeimage_swapmulti.py similarity index 95% rename from test_wholeimage_swapmutil.py rename to test_wholeimage_swapmulti.py index 9c49934..6cd2d53 100644 --- a/test_wholeimage_swapmutil.py +++ b/test_wholeimage_swapmulti.py @@ -8,7 +8,7 @@ import torch.nn.functional as F from torchvision import transforms from models.models import create_model from options.test_options import TestOptions -from insightface_func.face_detect_crop_mutil import Face_detect_crop +from insightface_func.face_detect_crop_multi import Face_detect_crop from util.reverse2original import reverse2wholeimage import os from util.add_watermark import watermark_image @@ -74,7 +74,7 @@ if __name__ == '__main__': 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'),opt.no_simswaplogo) + reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmulti.jpg'),opt.no_simswaplogo) print(' ') print('************ Done ! ************') diff --git a/test_wholeimage_swapspecific.py b/test_wholeimage_swapspecific.py index 7f85204..0c3f1b4 100644 --- a/test_wholeimage_swapspecific.py +++ b/test_wholeimage_swapspecific.py @@ -8,7 +8,7 @@ import torch.nn.functional as F from torchvision import transforms from models.models import create_model from options.test_options import TestOptions -from insightface_func.face_detect_crop_mutil import Face_detect_crop +from insightface_func.face_detect_crop_multi import Face_detect_crop from util.reverse2original import reverse2wholeimage import os from util.add_watermark import watermark_image diff --git a/util/videoswap_mutilspecific.py b/util/videoswap_multispecific.py similarity index 100% rename from util/videoswap_mutilspecific.py rename to util/videoswap_multispecific.py