diff --git a/inference_video_parallel.py b/inference_video_parallel.py index 44578c9..00cdeaa 100644 --- a/inference_video_parallel.py +++ b/inference_video_parallel.py @@ -81,7 +81,7 @@ def clear_buffer(user_args, buffer): for i in buffer: if user_args.png: cv2.imwrite('output/{:0>7d}.png'.format(cnt), i) - cnt += i + cnt += 1 else: vid_out.write(i)