CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-08-31 00:10:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

Debug path

Browse Source
This commit is contained in:
henryruhs
2025-03-11 14:43:10 +01:00
parent 94ad33cb1e
commit 866019d44f
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/tests/test_networks.py
+7 -7
View File
@@ -7,8 +7,8 @@ import torch
package_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) package_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(package_directory) sys.path.append(package_directory)
from face_swapper.src.networks.aad import AAD #from face_swapper.src.networks.aad import AAD
from face_swapper.src.networks.unet import UNet #from face_swapper.src.networks.unet import UNet
@pytest.mark.parametrize('output_size', [ 256 ]) @pytest.mark.parametrize('output_size', [ 256 ])
@@ -20,14 +20,14 @@ def test_aad_with_unet(output_size : int) -> None:
output_channels = 8192 output_channels = 8192
num_blocks = 2 num_blocks = 2
generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval() #generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval()
encoder = UNet(output_size).eval() #encoder = UNet(output_size).eval()
source_tensor = torch.randn(1, 512) source_tensor = torch.randn(1, 512)
target_tensor = torch.randn(1, 3, output_size, output_size) target_tensor = torch.randn(1, 3, output_size, output_size)
target_attributes = encoder(target_tensor) #target_attributes = encoder(target_tensor)
output_tensor = generator(source_tensor, target_attributes) #output_tensor = generator(source_tensor, target_attributes)
assert package_directory == None assert package_directory == None
assert output_tensor.shape == (1, 3, output_size, output_size) #assert output_tensor.shape == (1, 3, output_size, output_size)