diff --git a/scripts/core/txt2vid.py b/scripts/core/txt2vid.py index 40c2551..0cce6ef 100644 --- a/scripts/core/txt2vid.py +++ b/scripts/core/txt2vid.py @@ -200,4 +200,6 @@ def start_process(*args): warped_styled_frame_ = gr.Image.update() processed_frame = gr.Image.update() + # print('TOTAL TIME:', int(time.time() - processing_start_time)) + yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False) \ No newline at end of file diff --git a/scripts/core/vid2vid.py b/scripts/core/vid2vid.py index e4a202e..207af30 100644 --- a/scripts/core/vid2vid.py +++ b/scripts/core/vid2vid.py @@ -189,6 +189,7 @@ def start_process(*args): alpha_mask = alpha_mask + sdcn_anim_tmp.prev_frame_alpha_mask * 0.5 sdcn_anim_tmp.prev_frame_alpha_mask = alpha_mask + # alpha_mask = np.round(alpha_mask * 8) / 8 #> 0.3 alpha_mask = np.clip(alpha_mask, 0, 1) occlusion_mask = np.clip(alpha_mask * 255, 0, 255).astype(np.uint8)