diff --git a/python/pyproject.toml b/python/pyproject.toml index 465ad9a..3ef52ff 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "robologs-ros-utils" -version = "0.1.1a51" +version = "0.1.1a52" description = "robologs-ros-utils is an open source library of containerized data transformations for the robotics and drone communities" authors = ["roboto.ai "] license = "Apache-2.0" diff --git a/python/robologs_ros_utils/sources/ros1/ros_img_tools.py b/python/robologs_ros_utils/sources/ros1/ros_img_tools.py index 241b059..2d08039 100644 --- a/python/robologs_ros_utils/sources/ros1/ros_img_tools.py +++ b/python/robologs_ros_utils/sources/ros1/ros_img_tools.py @@ -66,7 +66,15 @@ def create_video_from_images(input_path, output_path, output_name="video.mp4", f output_video_path = os.path.join(output_path, output_name) output_video_path_temp = os.path.join(output_path, "temp.mp4") - for filename in sorted(glob.glob(os.path.join(input_path, "./*.jpg"))): + img_list = sorted(glob.glob(os.path.join(input_path, "./*.jpg"))) + + if not img_list: + sorted(glob.glob(os.path.join(input_path, "./*.png"))) + + if not img_list: + return + + for filename in img_list: img = cv2.imread(filename) if resize: img = cv2.resize(img, (0, 0), fx=resize, fy=resize, interpolation=cv2.INTER_LANCZOS4)