From f2977f9431945fbd748433a035fb14ca2747d7f8 Mon Sep 17 00:00:00 2001 From: hzwer <598460606@163.com> Date: Tue, 17 Nov 2020 18:32:07 +0800 Subject: [PATCH] Fix 4 road parallel --- inference_mp4_4x_parallel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/inference_mp4_4x_parallel.py b/inference_mp4_4x_parallel.py index f4281d6..76df6cf 100644 --- a/inference_mp4_4x_parallel.py +++ b/inference_mp4_4x_parallel.py @@ -87,8 +87,8 @@ while success: mid2 = model.inference(mid1, I1) mid0 = (((mid0 * 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8') mid1 = (((mid1 * 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8') - mid2 = (((mid2* 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8') - writeframe(p, mid0, mid1, mid) + mid2 = (((mid2 * 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8') + writeframe(p, mid0, mid1, mid2) pbar.update(4) img_list = img_list[-1] pbar.close()