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

Delete unused file

Browse Source
This commit is contained in:
henryruhs
2025-01-14 12:04:28 +01:00
parent 9e1c71b498
commit e3273221c9
1 changed files with 0 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/augmentations.py
-27
View File
@@ -1,27 +0,0 @@
import torch
from torch import Tensor
def apply_random_motion_blur(tensor_image : Tensor) -> Tensor:
kernel_size = 9
kernel = torch.zeros((kernel_size, kernel_size), dtype=torch.float32)
random_angle = torch.empty(1).uniform_(-2 * torch.pi, 2 * torch.pi)
dx = torch.cos(random_angle)
dy = torch.sin(random_angle)
center = kernel_size // 2
for i in range(kernel_size):
x = int(center + (i - center) * dx)
y = int(center + (i - center) * dy)
if 0 <= x < kernel_size and 0 <= y < kernel_size:
kernel[y, x] = 1
kernel /= kernel.sum()
kernel = kernel.unsqueeze(0).unsqueeze(0)
blurred_channels = []
for channel in tensor_image:
channel = channel.unsqueeze(0).unsqueeze(0)
channel = torch.nn.functional.conv2d(channel, kernel, padding=kernel_size // 2)
channel = channel.squeeze(0).squeeze(0)
blurred_channels.append(channel)
return torch.stack(blurred_channels)