github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 02:37:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added fanseg for future WF segmentation model

Browse source
This commit is contained in:
Colombo 2020-03-08 00:49:12 +04:00
commit 143792fd31
6 changed files with 429 additions and 112 deletions
Download patch file Download diff file Expand all files Collapse all files

7
core/leras/models.py
View file

@ -9,6 +9,7 @@ def initialize_models(nn):
def __init__(self, *args, name=None, **kwargs):
super().__init__(name=name)
self.layers = []
self.layers_by_name = {}
self.built = False
self.args = args
self.kwargs = kwargs
@ -31,7 +32,8 @@ def initialize_models(nn):
layer.build()
self.layers.append (layer)
self.layers_by_name[layer.name] = layer
def xor_list(self, lst1, lst2):
return [value for value in lst1+lst2 if (value not in lst1) or (value not in lst2) ]
@ -76,6 +78,9 @@ def initialize_models(nn):
weights += layer.get_weights()
return weights
def get_layer_by_name(self, name):
return self.layers_by_name.get(name, None)
def get_layers(self):
if not self.built:
self.build()