mirror of
https://github.com/zebrajr/faceswap.git
synced 2025-12-06 12:20:27 +01:00
6a3b674bef
* Remove tensorflow_probability requirement * setup.py - fix progress bars * requirements.txt: Remove pre python 3.9 packages * update apple requirements.txt * update INSTALL.md * Remove python<3.9 code * setup.py - fix Windows Installer * typing: python3.9 compliant * Update pytest and readthedocs python versions * typing fixes * Python Version updates - Reduce max version to 3.10 - Default to 3.10 in installers - Remove incompatible 3.11 tests * Update dependencies * Downgrade imageio dep for Windows * typing: merge optional unions and fixes * Updates - min python version 3.10 - typing to python 3.10 spec - remove pre-tf2.10 code - Add conda tests * train: re-enable optimizer saving * Update dockerfiles * Update setup.py - Apple Conda deps to setup.py - Better Cuda + dependency handling * bugfix: Patch logging to prevent Autograph errors * Update dockerfiles * Setup.py - Setup.py - stdout to utf-8 * Add more OSes to github Actions * suppress mac-os end to end test
77 lines
2.8 KiB
Python
77 lines
2.8 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_adabelief(dummy): # pylint:disable=unused-argument
|
|
""" Test for custom Adam optimizer """
|
|
_test_optimizer(optimizers.AdaBelief(), target=0.20)
|