CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Mask typing and naming related updates

Browse Source
This commit is contained in:
henryruhs
2025-03-14 08:13:21 +01:00
parent 5234874bc7
commit 33d00ac941
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/networks/masknet.py
+2 -4
View File
@@ -14,7 +14,7 @@ class MaskNet(nn.Module):
self.config_num_filters = config_parser.getint('training.model.masker', 'num_filters')
self.down_samples = self.create_down_samples(self.config_input_channels, self.config_num_filters)
self.up_samples = self.create_up_samples(self.config_num_filters)
self.bottleneck = BottleNeck(self.config_num_filters * 4)
self.bottleneck = BottleNeck(self.config_num_filters * 2)
self.conv = nn.Conv2d(self.config_num_filters, self.config_output_channels, kernel_size = 1)
self.sigmoid = nn.Sigmoid()
@@ -23,15 +23,13 @@ class MaskNet(nn.Module):
return nn.ModuleList(
[
DownSample(input_channels, num_filters),
DownSample(num_filters, num_filters * 2),
DownSample(num_filters, num_filters * 4)
DownSample(num_filters, num_filters * 2)
])
@staticmethod
def create_up_samples(num_filters : int) -> nn.ModuleList:
return nn.ModuleList(
[
UpSample(num_filters * 4, num_filters),
UpSample(num_filters * 2, num_filters),
UpSample(num_filters, num_filters)
])