neuralchen / SimSwap

An arbitrary face-swapping framework on images and videos with one single trained model!
Other
4.46k stars 882 forks source link

RTX 3060 #62

Closed shadow202020 closed 3 years ago

shadow202020 commented 3 years ago

(simswap) G:\Sim-swap>python test_video_swapsingle.py --isTrain false --use_mask --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path ./demo_file/Iron_man.jpg --video_path ./demo_file/multi_people_1080p.mp4 --output_path ./output/multi_test_swapsingle.mp4 --temp_path ./temp_results ------------ Options ------------- Arc_path: arcface_model/arcface_checkpoint.tar aspect_ratio: 1.0 batchSize: 8 checkpoints_dir: ./checkpoints cluster_path: features_clustered_010.npy data_type: 32 dataroot: ./datasets/cityscapes/ display_winsize: 512 engine: None export_onnx: None feat_num: 3 fineSize: 512 fp16: False gpu_ids: [0] how_many: 50 id_thres: 0.03 image_size: 224 input_nc: 3 instance_feat: False isTrain: False label_feat: False label_nc: 0 latent_size: 512 loadSize: 1024 load_features: False local_rank: 0 max_dataset_size: inf model: pix2pixHD multisepcific_dir: ./demo_file/multispecific nThreads: 2 n_blocks_global: 6 n_blocks_local: 3 n_clusters: 10 n_downsample_E: 4 n_downsample_global: 3 n_local_enhancers: 1 name: people nef: 16 netG: global ngf: 64 niter_fix_global: 0 no_flip: False no_instance: False no_simswaplogo: False norm: batch norm_G: spectralspadesyncbatch3x3 ntest: inf onnx: None output_nc: 3 output_path: ./output/multi_test_swapsingle.mp4 phase: test pic_a_path: ./demo_file/Iron_man.jpg pic_b_path: ./crop_224/zrf.jpg pic_specific_path: ./crop_224/zrf.jpg resize_or_crop: scale_width results_dir: ./results/ semantic_nc: 3 serial_batches: False temp_path: ./temp_results tf_log: False use_dropout: False use_encoded_image: False use_mask: True verbose: False video_path: ./demo_file/multi_people_1080p.mp4 which_epoch: latest -------------- End ---------------- input mean and std: 127.5 127.5 find model: ./insightface_func/models\antelope\glintr100.onnx recognition find model: ./insightface_func/models\antelope\scrfd_10g_bnkps.onnx detection set det-size: (640, 640) (142, 366, 4) Traceback (most recent call last): File "test_video_swapsingle.py", line 72, in no_simswaplogo=opt.no_simswaplogo,use_mask=opt.use_mask) File "G:\Sim-swap\util\videoswap.py", line 55, in video_swap net.load_state_dict(torch.load(save_pth)) File "C:\Users\mateu\anaconda3\envs\simswap\lib\site-packages\torch\serialization.py", line 581, in load with _open_file_like(f, 'rb') as opened_file: File "C:\Users\mateu\anaconda3\envs\simswap\lib\site-packages\torch\serialization.py", line 230, in _open_file_like return _open_file(name_or_buffer, mode) File "C:\Users\mateu\anaconda3\envs\simswap\lib\site-packages\torch\serialization.py", line 211, in init super(_open_file, self).init(open(name, mode)) FileNotFoundError: [Errno 2] No such file or directory: './parsing_model/checkpoint\79999_iter.pth'

woctezuma commented 3 years ago

Same issue as https://github.com/neuralchen/SimSwap/issues/14#issuecomment-882444218

The solution consists in:

We use the face parsing from face-parsing.PyTorch for image postprocessing. Please download the relative file and place it in ./parsing_model/checkpoint from this link.

https://github.com/neuralchen/SimSwap/blob/main/docs/guidance/preparation.md#installation

shadow202020 commented 3 years ago

It works, thanks a lot. I must have been missed this step during setup.

woctezuma commented 3 years ago

It was added recently to the set-up, so that is understandable!