fix face disc

NVIDIA · Sep 20, 2018 · ab25dd9 · ab25dd9
1 parent 0adb7b0
commit ab25dd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/vid2vid_model_D.py b/models/vid2vid_model_D.py
@@ -216,7 +216,7 @@ def forward(self, scale_T, tensors_list):
     def get_face_region(self, real_A):
         _, _, h, w = real_A.size()
         if not self.opt.openpose_only:
-            face = (real_A[:,2] > 0.9).nonzero()
+            face = (real_A[:,2] > 0.95).nonzero()
         else:            
             face = (((real_A[:,0] == 0.6) | (real_A[:,0] == 0.2)) & (real_A[:,1] == 0) & (real_A[:,2] == 0.6)).nonzero()
         if face.size()[0]: