From 9d186aec161c9a0959ce43edd7a21694d54bc478 Mon Sep 17 00:00:00 2001
From: NNNNAI <844294823@qq.com>
Date: Mon, 21 Jun 2021 11:04:39 +0800
Subject: [PATCH] Script for image face swapping

---
 test_wholeimage_swapsingle.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/test_wholeimage_swapsingle.py b/test_wholeimage_swapsingle.py
index 8ac51d3..b167a18 100644
--- a/test_wholeimage_swapsingle.py
+++ b/test_wholeimage_swapsingle.py
@@ -53,9 +53,7 @@ if __name__ == '__main__':
     #create latent id
     img_id_downsample = F.interpolate(img_id, scale_factor=0.5)
     latend_id = model.netArc(img_id_downsample)
-    latend_id = latend_id.detach().to('cpu')
-    latend_id = latend_id/np.linalg.norm(latend_id,axis=1,keepdims=True)
-    latend_id = latend_id.to('cuda')
+    latend_id = F.normalize(latend_id, p=2, dim=1)
 
 
     ############## Forward Pass ######################