From b18e54c3970f39c24fa346edbd5bf8332846e62c Mon Sep 17 00:00:00 2001
From: Wlad <9556979+gosticks@users.noreply.github.com>
Date: Sun, 21 Feb 2021 23:57:42 +0100
Subject: [PATCH] fix: thumbnail rendering

---
 utils/video.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/utils/video.py b/utils/video.py
index 79cb2ed..2517989 100644
--- a/utils/video.py
+++ b/utils/video.py
@@ -9,7 +9,7 @@ import numpy as np
 from scipy import interpolate
 
 
-def make_video(images, video_name: str, fps=5, ext: str = "mp4", post_process_frame=None):
+def make_video(images, video_name: str, fps=30, ext: str = "mp4", post_process_frame=None):
     images = np.array(images)
     width = images.shape[2]
     height = images.shape[1]
@@ -48,6 +48,7 @@ def save_to_video(
     config: object,
     fps=30,
     include_thumbnail=True,
+    thumbnail_size=0.25,
     dataset: SMPLyDataset = None,
     interpolation_target=None
 ):
@@ -101,11 +102,18 @@ def save_to_video(
         if interpolation_target is not None:
             # account for possible interpolation
             frame_idx = int(idx / inter_ratio)
-        overlay = cv2.imread(dataset.get_image_path(frame_idx))
+        img_path = dataset.get_image_path(frame_idx)
+        overlay = cv2.imread(img_path)
+
+        if overlay is None:
+            print("[error] image could not be ", img_path)
+            return img
+
         overlay = cv2.resize(
             overlay,
             dsize=(
-                int(overlay.shape[1] * 0.25), int(overlay.shape[0] * 0.25)
+                int(overlay.shape[1] * thumbnail_size),
+                int(overlay.shape[0] * thumbnail_size)
             ))
         img[0:overlay.shape[0], 0:overlay.shape[1]] = overlay
         return img