瀏覽代碼

Script for image face swapping

NNNNAI 4 年之前
父節點
當前提交
c964d806e7
共有 1 個文件被更改,包括 1 次插入3 次删除
  1. 1 3
      test_wholeimage_swapmutil.py

+ 1 - 3
test_wholeimage_swapmutil.py

@@ -54,9 +54,7 @@ if __name__ == '__main__':
     #create latent id
     #create latent id
     img_id_downsample = F.interpolate(img_id, scale_factor=0.5)
     img_id_downsample = F.interpolate(img_id, scale_factor=0.5)
     latend_id = model.netArc(img_id_downsample)
     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 ######################
     ############## Forward Pass ######################