mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-08 20:13:52 -04:00
03f5c671bc
* Remove PlaidML reference from readme files * Remove AMD option from installers * remove amd requirements and update setup.py * remove plaidml test from CI workflow * gpustats: remove plaidml backend * plaid removals: - faceswap.py - python version check - setup.cfg - plaidml typing ignore - lib.keras_utils - All plaid code - lib.launcher.py - All plaidml checks and configuration * remove tf2.2 specific code from GUI event reader * lib.model - remove all plaidml implementations * plugins.extract - remove plaidml code * plugins.train remove plaidml code * lib.convert - remove plaidml code * tools.model: remove plaidml code * Remove plaidML tests from unit tests * remove plaidml_utils and docsting cleanups * Remove plaidML refs from configs * fix keras imports
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
""" Tests for Faceswap Initializers.
|
|
|
|
Adapted from Keras tests.
|
|
"""
|
|
import pytest
|
|
|
|
import numpy as np
|
|
from numpy.testing import assert_allclose
|
|
# Ignore linting errors from Tensorflow's thoroughly broken import system
|
|
from tensorflow.keras import optimizers as k_optimizers # pylint:disable=import-error
|
|
from tensorflow.keras.layers import Dense, Activation # pylint:disable=import-error
|
|
from tensorflow.keras.models import Sequential # pylint:disable=import-error
|
|
|
|
from lib.model import optimizers
|
|
from lib.utils import get_backend
|
|
|
|
from tests.utils import generate_test_data, to_categorical
|
|
|
|
|
|
def get_test_data():
|
|
""" Obtain randomized test data for training """
|
|
np.random.seed(1337)
|
|
(x_train, y_train), _ = generate_test_data(num_train=1000,
|
|
num_test=200,
|
|
input_shape=(10,),
|
|
classification=True,
|
|
num_classes=2)
|
|
y_train = to_categorical(y_train)
|
|
return x_train, y_train
|
|
|
|
|
|
def _test_optimizer(optimizer, target=0.75):
|
|
x_train, y_train = get_test_data()
|
|
|
|
model = Sequential()
|
|
model.add(Dense(10, input_shape=(x_train.shape[1],)))
|
|
model.add(Activation("relu"))
|
|
model.add(Dense(y_train.shape[1]))
|
|
model.add(Activation("softmax"))
|
|
model.compile(loss="categorical_crossentropy",
|
|
optimizer=optimizer,
|
|
metrics=["accuracy"])
|
|
|
|
history = model.fit(x_train, y_train, epochs=2, batch_size=16, verbose=0)
|
|
assert history.history["accuracy"][-1] >= target
|
|
config = k_optimizers.serialize(optimizer)
|
|
optim = k_optimizers.deserialize(config)
|
|
new_config = k_optimizers.serialize(optim)
|
|
config["class_name"] = config["class_name"].lower()
|
|
new_config["class_name"] = new_config["class_name"].lower()
|
|
assert config == new_config
|
|
|
|
# Test constraints.
|
|
model = Sequential()
|
|
dense = Dense(10,
|
|
input_shape=(x_train.shape[1],),
|
|
kernel_constraint=lambda x: 0. * x + 1.,
|
|
bias_constraint=lambda x: 0. * x + 2.,)
|
|
model.add(dense)
|
|
model.add(Activation("relu"))
|
|
model.add(Dense(y_train.shape[1]))
|
|
model.add(Activation("softmax"))
|
|
model.compile(loss="categorical_crossentropy",
|
|
optimizer=optimizer,
|
|
metrics=["accuracy"])
|
|
model.train_on_batch(x_train[:10], y_train[:10])
|
|
kernel, bias = dense.get_weights()
|
|
assert_allclose(kernel, 1.)
|
|
assert_allclose(bias, 2.)
|
|
|
|
|
|
@pytest.mark.parametrize("dummy", [None], ids=[get_backend().upper()])
|
|
def test_adam(dummy): # pylint:disable=unused-argument
|
|
""" Test for custom Adam optimizer """
|
|
_test_optimizer(k_optimizers.Adam(), target=0.45) # pylint:disable=no-member
|
|
_test_optimizer(k_optimizers.Adam(decay=1e-3), target=0.45) # pylint:disable=no-member
|
|
|
|
|
|
@pytest.mark.parametrize("dummy", [None], ids=[get_backend().upper()])
|
|
def test_adabelief(dummy): # pylint:disable=unused-argument
|
|
""" Test for custom Adam optimizer """
|
|
_test_optimizer(optimizers.AdaBelief(), target=0.20)
|