mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-08 20:13:52 -04:00
815c843f63
* Add simple backend tests for lib.model * Document lib.model * Fix GMSD Loss for AMD * Remove obsolete code from lib.model
107 lines
2 KiB
ReStructuredText
Executable file
107 lines
2 KiB
ReStructuredText
Executable file
model package
|
|
=============
|
|
|
|
The Model Package handles interfacing with the neural network backend and holds custom objects.
|
|
|
|
.. contents:: Contents
|
|
:local:
|
|
|
|
|
|
model.initializers module
|
|
-------------------------
|
|
|
|
.. rubric:: Module Summary
|
|
|
|
.. autosummary::
|
|
:nosignatures:
|
|
|
|
~lib.model.initializers.ConvolutionAware
|
|
~lib.model.initializers.ICNR
|
|
|
|
.. automodule:: lib.model.initializers
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|
|
model.layers module
|
|
-------------------
|
|
|
|
.. rubric:: Module Summary
|
|
|
|
.. autosummary::
|
|
:nosignatures:
|
|
|
|
~lib.model.layers.GlobalMinPooling2D
|
|
~lib.model.layers.GlobalStdDevPooling2D
|
|
~lib.model.layers.L2_normalize
|
|
~lib.model.layers.PixelShuffler
|
|
~lib.model.layers.ReflectionPadding2D
|
|
~lib.model.layers.SubPixelUpscaling
|
|
|
|
.. automodule:: lib.model.layers
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|
|
model.losses module
|
|
-------------------
|
|
|
|
.. rubric:: Module Summary
|
|
|
|
.. autosummary::
|
|
:nosignatures:
|
|
|
|
~lib.model.losses.DSSIMObjective
|
|
~lib.model.losses.PenalizedLoss
|
|
~lib.model.losses.gaussian_blur
|
|
~lib.model.losses.generalized_loss
|
|
~lib.model.losses.gmsd_loss
|
|
~lib.model.losses.gradient_loss
|
|
~lib.model.losses.l_inf_norm
|
|
~lib.model.losses.mask_loss_wrapper
|
|
~lib.model.losses.scharr_edges
|
|
|
|
.. automodule:: lib.model.losses
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|
|
model.nn_blocks module
|
|
----------------------
|
|
|
|
.. automodule:: lib.model.nn_blocks
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|
|
model.normalization module
|
|
--------------------------
|
|
|
|
.. rubric:: Module Summary
|
|
|
|
.. autosummary::
|
|
:nosignatures:
|
|
|
|
~lib.model.normalization.InstanceNormalization
|
|
|
|
.. automodule:: lib.model.normalization
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|
|
model.optimizers module
|
|
-----------------------
|
|
|
|
.. automodule:: lib.model.optimizers
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|
|
model.session module
|
|
---------------------
|
|
|
|
.. automodule:: lib.model.session
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|
|
|