|
|
|
|
@@ -4,6 +4,8 @@
|
|
|
|
|
|
|
|
|
|
**Our method can realize **arbitrary face swapping** on images and videos with **one single trained model**.**
|
|
|
|
|
|
|
|
|
|
***We are recruiting full-time engineers. If you are interested, please send an [email](mailto:chen19910528@sjtu.edu.cn?subject=[GitHub]%20Source%20Han%20Sans) to my team. Please refer to the website for specific recruitment conditions: [Requirements](https://join.sjtu.edu.cn/Admin/QsPreview.aspx?qsid=44f5413a90974114b8f5e643177ef32d)***
|
|
|
|
|
|
|
|
|
|
Training and test code are now available!
|
|
|
|
|
[ <a href="https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/train.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a>](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/train.ipynb)
|
|
|
|
|
|
|
|
|
|
@@ -34,6 +36,8 @@ If you find this project useful, please star it. It is the greatest appreciation
|
|
|
|
|
|
|
|
|
|
## Top News <img width=8% src="./docs/img/new.gif"/>
|
|
|
|
|
|
|
|
|
|
**`2023-09-26`**: We fixed bugs in colab!
|
|
|
|
|
|
|
|
|
|
**`2023-04-25`**: We fixed the "AttributeError: 'SGD' object has no attribute 'defaults' now" bug. If you have already downloaded **arcface_checkpoint.tar**, please **download it again**. Also, you also need to update the scripts in ```./models/```.
|
|
|
|
|
|
|
|
|
|
**`2022-04-21`**: For resource limited users, we provide the cropped VGGFace2-224 dataset [[Google Driver] VGGFace2-224 (10.8G)](https://drive.google.com/file/d/19pWvdEHS-CEG6tW3PdxdtZ5QEymVjImc/view?usp=sharing) [[Baidu Driver]](https://pan.baidu.com/s/1OiwLJHVBSYB4AY2vEcfN0A) [Password: lrod].
|
|
|
|
|
@@ -70,6 +74,7 @@ If you find this project useful, please star it. It is the greatest appreciation
|
|
|
|
|
- imageio
|
|
|
|
|
- moviepy
|
|
|
|
|
- insightface
|
|
|
|
|
- ***timm==0.5.4***
|
|
|
|
|
|
|
|
|
|
## Training
|
|
|
|
|
|
|
|
|
|
@@ -92,7 +97,7 @@ For faster convergence and better results, a large batch size (more than 16) is
|
|
|
|
|
python train.py --name simswap224_test --batchSize 8 --gpu_ids 0 --dataset /path/to/VGGFace2HQ --Gdeep False
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
[Colab demo for training 224 model](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/train.ipynb)
|
|
|
|
|
[Colab demo for training 224 model][ <a href="https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/train.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a>](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/train.ipynb)
|
|
|
|
|
|
|
|
|
|
For faster convergence and better results, a large batch size (more than 16) is recommended!
|
|
|
|
|
|
|
|
|
|
@@ -112,7 +117,7 @@ python train.py --name simswap512_test --batchSize 16 --gpu_ids 0 --dataset /p
|
|
|
|
|
|
|
|
|
|
<div style="background: yellow; width:140px; font-weight:bold;font-family: sans-serif;">Stronger feature</div>
|
|
|
|
|
|
|
|
|
|
[Colab for switching specific faces in multi-face videos](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/MultiSpecific.ipynb)
|
|
|
|
|
[Colab for switching specific faces in multi-face videos][ <a href="https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/MultiSpecific.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a>](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/MultiSpecific.ipynb)
|
|
|
|
|
|
|
|
|
|
[Image face swapping demo & Docker image on Replicate](https://replicate.ai/neuralchen/simswap-image)
|
|
|
|
|
|
|
|
|
|
@@ -166,7 +171,7 @@ At the same time, if you have suggestions for our project, please feel free to a
|
|
|
|
|
For academic and non-commercial use only.The whole project is under the CC-BY-NC 4.0 license. See [LICENSE](https://github.com/neuralchen/SimSwap/blob/main/LICENSE) for additional details.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## To cite our paper
|
|
|
|
|
## To cite our papers
|
|
|
|
|
```
|
|
|
|
|
@inproceedings{DBLP:conf/mm/ChenCNG20,
|
|
|
|
|
author = {Renwang Chen and
|
|
|
|
|
@@ -178,6 +183,22 @@ For academic and non-commercial use only.The whole project is under the CC-BY-NC
|
|
|
|
|
year = {2020}
|
|
|
|
|
}
|
|
|
|
|
```
|
|
|
|
|
```
|
|
|
|
|
@Article{simswapplusplus,
|
|
|
|
|
author = {Xuanhong Chen and
|
|
|
|
|
Bingbing Ni and
|
|
|
|
|
Yutian Liu and
|
|
|
|
|
Naiyuan Liu and
|
|
|
|
|
Zhilin Zeng and
|
|
|
|
|
Hang Wang},
|
|
|
|
|
title = {SimSwap++: Towards Faster and High-Quality Identity Swapping},
|
|
|
|
|
journal = {{IEEE} Trans. Pattern Anal. Mach. Intell.},
|
|
|
|
|
volume = {46},
|
|
|
|
|
number = {1},
|
|
|
|
|
pages = {576--592},
|
|
|
|
|
year = {2024}
|
|
|
|
|
}
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
## Related Projects
|
|
|
|
|
|
|
|
|
|
|