From 6bd11afe4d42eda0a8c3ecfac1c1260e9825e083 Mon Sep 17 00:00:00 2001 From: sinofis Date: Sat, 9 Jan 2021 18:36:03 +0100 Subject: [PATCH] face part training - fix --- models/Model_SAEHD/Model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 7f351c0..ac455d9 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -397,10 +397,10 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... # unpack masks from one combined mask gpu_target_srcm = tf.clip_by_value (gpu_target_srcm_all, 0, 1) gpu_target_dstm = tf.clip_by_value (gpu_target_dstm_all, 0, 1) - gpu_target_srcm_eye_mouth = tf.clip_by_value (gpu_target_srcm_all-1, 0, 1) - gpu_target_dstm_eye_mouth = tf.clip_by_value (gpu_target_dstm_all-1, 0, 1) - gpu_target_srcm_mouth = tf.clip_by_value (gpu_target_srcm_all-2, 0, 1) - gpu_target_dstm_mouth = tf.clip_by_value (gpu_target_dstm_all-2, 0, 1) + gpu_target_srcm_eye_mouth = tf.clip_by_value (gpu_target_srcm_em-1, 0, 1) + gpu_target_dstm_eye_mouth = tf.clip_by_value (gpu_target_dstm_em-1, 0, 1) + gpu_target_srcm_mouth = tf.clip_by_value (gpu_target_srcm_em-2, 0, 1) + gpu_target_dstm_mouth = tf.clip_by_value (gpu_target_dstm_em-2, 0, 1) gpu_target_srcm_eyes = tf.clip_by_value (gpu_target_srcm_eye_mouth-gpu_target_srcm_mouth, 0, 1) gpu_target_dstm_eyes = tf.clip_by_value (gpu_target_dstm_eye_mouth-gpu_target_dstm_mouth, 0, 1)