Merge branch 'main' of https://github.com/neuralchen/SimSwap into main

chi-cat · Jun 15, 2021 · faa6fe1 · faa6fe1
2 parents 7d68e1d + 5e3753b
commit faa6fe1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_one_image.py b/test_one_image.py
@@ -55,7 +55,7 @@ def lcm(a, b): return abs(a * b) / fractions.gcd(a, b) if a and b else 0
 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)
+latend_id = latend_id/np.linalg.norm(latend_id,axis=1,keepdims=True)
 latend_id = latend_id.to('cuda')