CalvinBackup/neuralchen-SimSwap
1
0
Fork 0
Code Issues 288 Pull Requests 16 Actions Packages Projects Releases 2 Wiki Activity

The arcface model cannot be loaded #51

New Issue
Open
opened 2021-07-16 05:28:16 +02:00 by alan525 · 3 comments
alan525 commented 2021-07-16 05:28:16 +02:00 (Migrated from github.com)
Copy Link
Copy Source

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？

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 2021-07-17 08:42:32 +02:00 (Migrated from github.com)
Copy Link
Copy Source

Could you share the command line you used?

Could you share the command line you used?
👀 1
alan525 commented 2021-07-18 13:56:07 +02:00 (Migrated from github.com)
Copy Link
Copy Source

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)

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 2021-07-18 15:58:56 +02:00 (Migrated from github.com)
Copy Link
Copy Source

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

#9, #10, #24

> SimSwap/models/fs_model.py line 64 > netArc_checkpoint=torch.load(netArc_checkpoint) #9, #10, #24
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
GIF
GUI
Increase inference time
A solution to increase inference time
 bug
Something isn't working
 documentation
Improvements or additions to documentation
 duplicate
This issue or pull request already exists
 enhancement
New feature or request
 good first issue
Good for newcomers
 help wanted
Extra attention is needed
 invalid
This doesn't seem right
 question
Further information is requested
 wontfix
This will not be worked on
No labels
Milestone
No items
No Milestone
Projects
Clear projects
No projects
Assignees
Clear assignees
CalvinBackup (Calvin Backup)
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: CalvinBackup/neuralchen-SimSwap#51