From 71c3820d2d8135a9644c13d17fd7d7617a20dbb1 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 21 Oct 2020 11:08:12 -0600 Subject: [PATCH] Fix process_video --- codes/process_video.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/codes/process_video.py b/codes/process_video.py index 056f57fa..24a6181e 100644 --- a/codes/process_video.py +++ b/codes/process_video.py @@ -161,11 +161,8 @@ if __name__ == "__main__": model.feed_data(data, need_GT=need_GT) model.test() + visuals = model.get_current_visuals()['rlt'] - if isinstance(model.fake_H, tuple): - visuals = model.fake_H[img_index].detach() - else: - visuals = model.fake_H.detach() if recurrent_mode: recurrent_entry = visuals visuals = visuals.cpu().float()