mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2024-12-25 23:41:12 -08:00
61472cdaf7
removed support of extracted(aligned) PNG faces. Use old builds to convert from PNG to JPG. fanseg model file in facelib/ is renamed
21 lines
548 B
Python
21 lines
548 B
Python
import numpy as np
|
|
from tensorflow.python.ops import init_ops
|
|
|
|
from core.leras import nn
|
|
|
|
tf = nn.tf
|
|
|
|
from .CA import CAInitializerSubprocessor
|
|
|
|
class initializers():
|
|
class ca (init_ops.Initializer):
|
|
def __call__(self, shape, dtype=None, partition_info=None):
|
|
return tf.zeros( shape, dtype=dtype, name="_cai_")
|
|
|
|
@staticmethod
|
|
def generate_batch( data_list, eps_std=0.05 ):
|
|
# list of (shape, np.dtype)
|
|
return CAInitializerSubprocessor (data_list).run()
|
|
|
|
nn.initializers = initializers
|