From 8a50718b2aff295dd65b51d119db1c72d3476d6a Mon Sep 17 00:00:00 2001 From: hzwer <598460606@163.com> Date: Sun, 15 Nov 2020 16:45:52 +0800 Subject: [PATCH] Fix montage --- inference_mp4_2x.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py index e7ccaeb..9df0dbd 100644 --- a/inference_mp4_2x.py +++ b/inference_mp4_2x.py @@ -47,13 +47,13 @@ while success: mid = model.inference(I0, I1) mid = ((mid[0].cpu().detach().numpy().transpose(1, 2, 0))*255.).astype('uint8') if args.montage: - output.write(torch.cat((lastframe, lastframe), 2)) - output.write(torch.cat((lastframe, mid[:h, :w]), 2)) + output.write(np.concatenate((lastframe, lastframe), 1)) + output.write(np.concatenate((lastframe, mid[:h, :w]), 1)) else: output.write(lastframe) output.write(mid[:h, :w]) if args.montage: - output.write(torch.cat((frame, frame), 2)) + output.write(np.concatenate((frame, frame), 1)) else: output.write(frame) output.release()