mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix process_video

Browse Source
This commit is contained in:
James Betker 2020-10-21 11:08:12 -06:00
parent 3c6e600e48
commit 71c3820d2d
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/process_video.py
View File

@ -161,11 +161,8 @@ if __name__ == "__main__":
model.feed_data(data, need_GT=need_GT) model.feed_data(data, need_GT=need_GT)
model.test() 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: if recurrent_mode:
recurrent_entry = visuals recurrent_entry = visuals
visuals = visuals.cpu().float() visuals = visuals.cpu().float()