github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 06:23:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

test

Browse source
This commit is contained in:
jh 2021-03-16 17:11:31 -07:00
commit 0d5b67a355
1 changed files with 16 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

34
models/Model_SAEHD/Model.py
View file

@ -497,32 +497,30 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
print("Shape (gpu_pred_src_src_d): ", gpu_pred_src_src_d.shape) print("Shape (gpu_pred_src_src_d): ", gpu_pred_src_src_d.shape)
print("Shape (gpu_pred_src_src_d2): ", gpu_pred_src_src_d2.shape) print("Shape (gpu_pred_src_src_d2): ", gpu_pred_src_src_d2.shape)
for i in gpu_pred_src_src_d2.shape:
print("Shape (gpu_pred_src_src_d2) i: ", i)
# def get_smooth_noisy_labels_like(label, tensor, smoothing=0.2, noise=0.05): def get_smooth_noisy_label(label, smoothing=0.2, noise=0.05):
# new_labels = [] if np.random.random() < noise:
# for label in labels: label = 0 if label == 1 else 1
# if np.random.random() < noise: if label == 0:
# label = 0 if label == 1 else 1 return lambda x: np.random.uniform(0, 0+smoothing/2)
# if label == 0: else:
# new_labels.append(np.random.uniform(0, 0+smoothing/2)) return lambda x: np.random.uniform(1-smoothing/2, 1.0)
# else:
# new_labels.append(np.random.uniform(1-smoothing/2, 1.0))
# return new_labels
gpu_pred_src_src_d_ones = tf.ones_like (gpu_pred_src_src_d) gpu_pred_src_src_d_ones = tf.map_fn (get_smooth_noisy_label(1, smoothing=0.2, noise=0.05), gpu_pred_src_src_d)
gpu_pred_src_src_d_zeros = tf.zeros_like(gpu_pred_src_src_d) gpu_pred_src_src_d_zeros = tf.map_fn (get_smooth_noisy_label(0, smoothing=0.2, noise=0.05), gpu_pred_src_src_d)
gpu_pred_src_src_d2_ones = tf.ones_like (gpu_pred_src_src_d2) print("Shape (gpu_pred_src_src_d_ones): ", gpu_pred_src_src_d.shape)
gpu_pred_src_src_d2_zeros = tf.zeros_like(gpu_pred_src_src_d2) print("Shape (gpu_pred_src_src_d_zeros): ", gpu_pred_src_src_d2.shape)
gpu_pred_src_src_d2_ones = tf.map_fn (get_smooth_noisy_label(1, smoothing=0.2, noise=0.05), gpu_pred_src_src_d2)
gpu_pred_src_src_d2_zeros = tf.map_fn (get_smooth_noisy_label(0, smoothing=0.2, noise=0.05), gpu_pred_src_src_d2)
gpu_target_src_d, \ gpu_target_src_d, \
gpu_target_src_d2 = self.D_src(gpu_target_src_masked_opt) gpu_target_src_d2 = self.D_src(gpu_target_src_masked_opt)
gpu_target_src_d_ones = tf.ones_like(gpu_target_src_d) gpu_target_src_d_ones = tf.map_fn (get_smooth_noisy_label(1, smoothing=0.2, noise=0.05), gpu_target_src_d)
gpu_target_src_d2_ones = tf.ones_like(gpu_target_src_d2) gpu_target_src_d2_ones = tf.map_fn (get_smooth_noisy_label(1, smoothing=0.2, noise=0.05), gpu_target_src_d2)
gpu_D_src_dst_loss = (DLoss(gpu_target_src_d_ones , gpu_target_src_d) + \ gpu_D_src_dst_loss = (DLoss(gpu_target_src_d_ones , gpu_target_src_d) + \
DLoss(gpu_pred_src_src_d_zeros , gpu_pred_src_src_d) ) * 0.5 + \ DLoss(gpu_pred_src_src_d_zeros , gpu_pred_src_src_d) ) * 0.5 + \