From 9e692ed4f495355fc50a51e72fd853f081191175 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Tue, 17 Nov 2020 19:10:08 +0800
Subject: [PATCH] Fixed parallel

---
 inference_mp4_4x_parallel.py | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/inference_mp4_4x_parallel.py b/inference_mp4_4x_parallel.py
index e30a3a0..9b43178 100644
--- a/inference_mp4_4x_parallel.py
+++ b/inference_mp4_4x_parallel.py
@@ -49,20 +49,21 @@ def writeframe(I0, mid0, mid1, mid2, I1, p):
             if skip_frame % 100 == 0:
                 print("Warning: Your video has {} static frames, skipping them may change the duration of the generated video.".format(skip_frame))
             skip_frame += 1
-    if args.png:
-        cv2.imwrite('output/{:0>7d}.png'.format(cnt), I0[i])        
-        cnt += 1
-        cv2.imwrite('output/{:0>7d}.png'.format(cnt), mid0[i])
-        cnt += 1
-        cv2.imwrite('output/{:0>7d}.png'.format(cnt), mid1[i])
-        cnt += 1
-        cv2.imwrite('output/{:0>7d}.png'.format(cnt), mid2[i])
-        cnt += 1
-    else:
-        output.write(I0[i])
-        output.write(mid0[i])
-        output.write(mid1[i])
-        output.write(mid2[i])
+            continue
+        if args.png:
+            cv2.imwrite('output/{:0>7d}.png'.format(cnt), I0[i])        
+            cnt += 1
+            cv2.imwrite('output/{:0>7d}.png'.format(cnt), mid0[i])
+            cnt += 1
+            cv2.imwrite('output/{:0>7d}.png'.format(cnt), mid1[i])
+            cnt += 1
+            cv2.imwrite('output/{:0>7d}.png'.format(cnt), mid2[i])
+            cnt += 1
+        else:
+            output.write(I0[i])
+            output.write(mid0[i])
+            output.write(mid1[i])
+            output.write(mid2[i])
 ph = ((h - 1) // 32 + 1) * 32
 pw = ((w - 1) // 32 + 1) * 32
 padding = (0, pw - w, 0, ph - h)