Fix process_video

2020-10-21 11:08:12 -06:00 · 2020-10-21 11:08:12 -06:00 · 71c3820d2d
commit 71c3820d2d
parent 3c6e600e48
1 changed files with 1 additions and 4 deletions
--- 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()