1
0
Fork 0
mirror of https://github.com/deepfakes/faceswap synced 2025-06-09 04:36:50 -04:00
Commit graph

9 commits

Author SHA1 Message Date
torzdf
43a4d06540
Smart Masks - Training Implementation (#914)
* Smart Masks - Training

- Reinstate smart mask training code
- Reinstate mask_type back to model.config
- change 'replicate_input_mask to 'learn_mask'
- Add learn mask option
- Add mask loading from alignments to plugins.train.trainer
- Add mask_blur and mask threshold options
- _base.py - Pass mask options through training_opts dict
- plugins.train.model - check for mask_type not None for learn_mask and penalized_mask_loss
- Limit alignments loading to just those faces that appear in the training folder
- Raise error if not all training images have an alignment, and alignment file is required
- lib.training_data - Mask generation code
- lib.faces_detect - cv2 dimension stripping bugfix
- Remove cv2 linting code

* Update mask helptext in cli.py

* Fix Warp to Landmarks
Remove SHA1 hashing from training data

* Update mask training config

* Capture missing masks at training init

* lib.image.read_image_batch - Return filenames with batch for ordering

* scripts.train - Documentation

* plugins.train.trainer - documentation

* Ensure backward compatibility.
Fix convert for new predicted masks

* Update removed masks to components for legacy models.
2019-12-05 16:02:01 +00:00
torzdf
70ee125283 Serialize masks to alignments file
- Add new serializers (npy + compressed)
- Remove Serializer option from cli
- Revert get_aligned call in scripts/extract
- Default alignments to compressed
- Size masks to 128px and compress
- Remove mask thresholding/blur from generation code
- Add Mask class to lib/faces_detect
- Revert debug landmarks to aligned face
- Revert non-extraction code to staging version
2019-10-13 22:50:28 +00:00
kvrooman
995a857154 Smart Mask Exposure for Extraction & Training (#831)
Smart Masks - Initial Commit
2019-10-07 16:16:18 +01:00
Artem Ivanov
c6319ca4f9 Model output size assertions fix (#816)
* added sanity check for RealFace output size to be divisible by 32
2019-07-29 23:30:13 +01:00
torzdf
40103b9c4a Model Structure Changes
Standardize inputs for AutoEncoders
Name outputs from AutoEncoders
Name layers in nn_blocks.py
2019-07-18 17:04:47 +00:00
torzdf
5418fba726
Add Convolutional Aware Initialization (#795)
* Training: Add Convolutional Aware Initialization config option

* Centralize Conv2D layer for handling initializer

* Add 'is-output' to NNMeta to indicate that network is an output to the Model
2019-07-16 10:09:29 +01:00
Artem Ivanov
568bd7048a RealFace Plugin: Encoder complexity fix (#782)
- fixed a bug that prevented altering default value of `complexity_encoder`
2019-07-07 12:19:08 +01:00
torzdf
0eff0a1719 Config Changes + Bugfixes
Bugfix: Fully disable keypress monitor for GUI
Bugfix: Preview - Handle missing alignments file
Config changes:
    - Separate plugin defaults into their own files
    - Move mask_type to global training config
    - Add ability to pass in custom config files
2019-06-11 01:13:42 +00:00
Artem Ivanov
e474dd1005 Pegasus model plug-in rc1 (#701)
* added new RealFace training model (codenamed: Pegasus)

* minor re-factoring (doesn't break compatibility)

* fixed tf error for input/output sizes with ratios other than 1:2

* linting

* Minor formatting/grammar ammends

* Make loss a fixed config item. Make learning rate an unfixed item

* Check input and output sizes
2019-04-14 19:28:56 +00:00