diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py index a759da6..8b50115 100644 --- a/inference_mp4_2x.py +++ b/inference_mp4_2x.py @@ -60,7 +60,8 @@ while success: p = (F.interpolate(I0, (16, 16), mode='bilinear', align_corners=False) - F.interpolate(I1, (16, 16), mode='bilinear', align_corners=False)).abs().mean() if p < 1e-3 and args.skip: - print("Warning: Your video has {} static frames, skipping them change the duration of the generated video.".format(cnt)) + if cnt % 10 == 0: + print("Warning: Your video has {} static frames, skipping them change the duration of the generated video.".format(cnt)) cnt += 1 pbar.update(1) continue @@ -74,9 +75,7 @@ while success: output.write(np.concatenate((lastframe, mid1[:h, :w]), 1)) else: output.write(lastframe) - output.write(mid0[:h, :w]) output.write(mid1[:h, :w]) - output.write(mid2[:h, :w]) pbar.update(1) if args.montage: output.write(np.concatenate((lastframe, lastframe), 1)) diff --git a/inference_mp4_4x.py b/inference_mp4_4x.py index cead491..7dfbd09 100644 --- a/inference_mp4_4x.py +++ b/inference_mp4_4x.py @@ -60,7 +60,8 @@ while success: p = (F.interpolate(I0, (16, 16), mode='bilinear', align_corners=False) - F.interpolate(I1, (16, 16), mode='bilinear', align_corners=False)).abs().mean() if p < 1e-3 and args.skip: - print("Warning: Your video has {} static frames, skipping them may change the duration of the generated video.".format(cnt)) + if cnt % 10 == 0: + print("Warning: Your video has {} static frames, skipping them may change the duration of the generated video.".format(cnt)) cnt += 1 pbar.update(1) continue