CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-06-25 07:59:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

ugly training code

Browse Source
This commit is contained in:
harisreedhar
2024-12-10 22:45:00 +05:30
committed by henryruhs
parent fe28b6fffe
commit 650268c06b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+2 -2
View File
@@ -318,9 +318,9 @@ class FaceSwapper(pytorch_lightning.LightningModule):
crop_vision_tensor = vision_tensor[:, :, crop_height : height - crop_height, crop_width : width - crop_width]
crop_vision_tensor = torch.nn.functional.interpolate(crop_vision_tensor, size = (112, 112), mode = 'bilinear')
crop_vision_tensor[:, :, :padding[0], :] = 0
crop_vision_tensor[:, :, -padding[1]:, :] = 0
crop_vision_tensor[:, :, 112 - padding[1]:, :] = 0
crop_vision_tensor[:, :, :, :padding[2]] = 0
crop_vision_tensor[:, :, :, -padding[3]:] = 0
crop_vision_tensor[:, :, :, 112 - padding[3]:] = 0
embedding = self.arcface(crop_vision_tensor)
embedding = torch.nn.functional.normalize(embedding, p = 2, dim = 1)
return embedding