SimSwapHD
SimSwapHD copied to clipboard
Does anyone successed on test_one_image.py ?
python test_one_image.py --image_size 224 --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/
error 1:
return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'models.arcface_models'
solution: copy arcface_models.py file to models dir
run it again, still an error:
self.netArc = netArc_checkpoint['model'].module TypeError: 'ResNet' object is not subscriptable
How to solve it ?