KeyError: 'features.conv4.weight'

Heonozis / FaceShifter-pytorch

115 stars 27 forks source link

KeyError: 'features.conv4.weight' #2

Closed wangchao0899 closed 4 years ago

wangchao0899 commented 4 years ago

Hi，when I test Demo，I got a Error~

File "D:\FaceShifter-pytorch\face_modules\mtcnn_pytorch\src\get_nets.py", line 102, in init p.data = torch.FloatTensor(weights[n]) KeyError: 'features.conv4.weight'

Heonozis commented 4 years ago

Let me upload the latest weights to cloud it seems weights are outdated

Heonozis commented 4 years ago

@wangchao0899 please refer updated https://github.com/Heonozis/FaceShifter-pytorch#demo section and see updated weights

wangchao0899 commented 4 years ago

@Heonozis Great！ It works~