CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-04-19 15:56:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Final refactoring for AAD done

Browse Source
This commit is contained in:
henryruhs
2025-02-24 00:18:35 +01:00
parent bc174186eb
commit 303cbfa024
1 changed files with 0 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/networks/aad.py
-18
View File
@@ -62,24 +62,6 @@ class AdaptiveFeatureModulation(nn.Module):
return primary_layers
@staticmethod
def _create_primary_layers(input_channels : int, output_channels : int, attribute_channels : int, identity_channels : int, num_blocks : int) -> nn.ModuleList:
primary_layers = nn.ModuleList()
for index in range(num_blocks):
primary_layers.extend(
[
FeatureModulation(input_channels, attribute_channels, identity_channels),
nn.ReLU(inplace = True)
])
if index < num_blocks - 1:
primary_layers.append(nn.Conv2d(input_channels, input_channels, kernel_size = 3, padding = 1, bias = False))
else:
primary_layers.append(nn.Conv2d(input_channels, output_channels, kernel_size = 3, padding = 1, bias = False))
return primary_layers
@staticmethod
def create_shortcut_layers(input_channels : int, output_channels : int, attribute_channels : int, identity_channels : int) -> nn.ModuleList:
shortcut_layers = nn.ModuleList()