search

neuralchen / SimSwap

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

The arcface model cannot be loaded #51

Open alan525 opened 3 years ago

alan525 commented 3 years ago

It's a great project. I follow the instructions to configure the environment. When the code runs to netArc_checkpoint=torch.load(netArc_checkpoint) an error is reported. ..torch/serialization.py line 702, in _legacy_load result = unpickler.load() AttributeError: Can't get attribute 'ArcMarginModel' on <module 'models' from 'SimSwap-main/models/init.py'> How to solve it? Is the model incorrect？

NNNNAI commented 3 years ago

Could you share the command line you used?

alan525 commented 3 years ago

command python test_one_image.py --isTrain false --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path crop_224/6.jpg --pic_b_path crop_224/ds.jpg --output_path output/

I'm sure the download model and path are correct.However, the model loading still failed. SimSwap/models/fs_model.py line 64 netArc_checkpoint=torch.load(netArc_checkpoint)

woctezuma commented 3 years ago

SimSwap/models/fs_model.py line 64 netArc_checkpoint=torch.load(netArc_checkpoint)

9, #10, #24