Spaces:

ygtxr1997
/

ReliableSwap_Demo

Running

gavinyuan commited on Jun 6, 2023

Commit

5c64773

1 Parent(s): afe1649

update: GPEN weights

Files changed (1) hide show

app.py CHANGED Viewed

@@ -308,7 +308,7 @@ def process_video(
         ''' Finetune paste masks '''
         target_facial_mask = trick.get_any_mask(target,
                                                 par=[1, 2, 3, 4, 5, 6, 10, 11, 12, 13]).squeeze()  # in [0,1]
-        target_facial_mask = target_facial_mask.cpu().numpy().astype(np.float)
         target_facial_mask = trick.finetune_mask(target_facial_mask, target_lmks)  # in [0,1]
         t_facial_masks.append(target_facial_mask)
         ''' Face swapping '''

         ''' Finetune paste masks '''
         target_facial_mask = trick.get_any_mask(target,
                                                 par=[1, 2, 3, 4, 5, 6, 10, 11, 12, 13]).squeeze()  # in [0,1]
+        target_facial_mask = target_facial_mask.cpu().numpy().astype(np.float32)
         target_facial_mask = trick.finetune_mask(target_facial_mask, target_lmks)  # in [0,1]
         t_facial_masks.append(target_facial_mask)
         ''' Face swapping '''