GithubStar/faceswap
1
0
Fork 0
mirror of https://github.com/deepfakes/faceswap.git synced 2026-10-16 21:13:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

extract bugfixes: vgg-obstructed model path + unet-dfl output shape

Browse Source
This commit is contained in:
torzdf
2026-05-21 12:02:26 +01:00
parent 815b11f104
commit 7aa85b29aa
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/extract/mask/unet_dfl.py
+1 -1
View File
@@ -68,7 +68,7 @@ class UNetDFL(FacePlugin):
-------
The predicted masks from the plugin
"""
return self.from_torch(batch.transpose(0, 3, 1, 2)).transpose(0, 2, 3, 1)
return self.from_torch(batch.transpose(0, 3, 1, 2))[:, 0]
class ConvBlock(nn.Module):
plugins/extract/mask/vgg_obstructed.py
+1 -1
View File
@@ -39,7 +39,7 @@ class VGGObstructed(FacePlugin):
-------
The loaded VGGObstructed model
"""
weights = GetModel("Nirkin_500_softmax_v2.pth", 8).model_path
weights = GetModel("Nirkin_500_softmax_v2.pth", 5).model_path
assert isinstance(weights, str)
return T.cast(VGGObstructedModel, self.load_torch_model(VGGObstructedModel(),
weights,