mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-06-25 07:59:55 +02:00
harisreedhar
108 lines
1.7 KiB
Markdown
108 lines
1.7 KiB
Markdown
Face Swapper
|
|
=================
|
|
|
|
> Swap one face over another face.
|
|
|
|
|
|
|
|
|
|
Preview
|
|
-------
|
|
|
|
![Preview]()
|
|
|
|
|
|
Installation
|
|
------------
|
|
|
|
```
|
|
pip install -r requirements.txt
|
|
```
|
|
|
|
|
|
Example
|
|
-------
|
|
|
|
This example utilizes the MegaFace dataset to train an ArcFace Converter for SimSwap.
|
|
|
|
```
|
|
[preparing.dataset]
|
|
dataset_path = datasets/train
|
|
folder_pattern = {}/*
|
|
image_pattern = {}/*.*g
|
|
same_person_probability = 0.2
|
|
|
|
[training.loader]
|
|
batch_size = 24
|
|
num_workers = 12
|
|
|
|
[training.model]
|
|
id_embedder_path = assets/models/id_embedder.pt
|
|
landmarker_path = assets/models/landmarker.pt
|
|
motion_extractor_path = assets/models/motion_extractor.pt
|
|
|
|
[training.model.generator]
|
|
num_blocks = 2
|
|
id_channels = 512
|
|
|
|
[training.model.discriminator]
|
|
input_channels = 3
|
|
num_filters = 64
|
|
num_layers = 5
|
|
num_discriminators = 3
|
|
kernel_size = 4
|
|
|
|
[training.losses]
|
|
weight_adversarial = 1
|
|
weight_id = 20
|
|
weight_attribute = 10
|
|
weight_reconstruction = 10
|
|
weight_pose = 100
|
|
|
|
[training.trainer]
|
|
max_epochs = 50
|
|
learning_rate = 0.0004
|
|
precision = 16-mixed
|
|
automatic_optimization = false
|
|
|
|
[training.output]
|
|
checkpoint_path = outputs/last.ckpt
|
|
directory_path = outputs
|
|
file_pattern = 'checkpoint-{epoch}-{step}-{l_G:.4f}-{l_D:.4f}'
|
|
preview_frequency = 250
|
|
validation_frequency = 1000
|
|
|
|
[exporting]
|
|
directory_path = export
|
|
source_path = outputs/last.ckpt
|
|
target_path = export/face_swapper.onnx
|
|
opset_version = 15
|
|
|
|
[inference]
|
|
generator_path = outputs/last.ckpt
|
|
id_embedder_path = assets/models/id_embedder.pt
|
|
source_path = assets/images/source.jpg
|
|
target_path = assets/models/target.jpg
|
|
output_path = outputs/output.jpg
|
|
```
|
|
|
|
|
|
Training
|
|
--------
|
|
|
|
Train the Face swapper model.
|
|
|
|
```
|
|
python train.py
|
|
```
|
|
|
|
|
|
Exporting
|
|
---------
|
|
|
|
Export the model to ONNX.
|
|
|
|
```
|
|
python export.py
|
|
```
|