From 9f2f738a7c62d99c9ea60f565d50ba88042bbf9e Mon Sep 17 00:00:00 2001 From: zhangmozhe Date: Mon, 29 Mar 2021 22:14:45 +0800 Subject: [PATCH] Update util.py --- utils/util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/util.py b/utils/util.py index 3f23a03..1fc324e 100755 --- a/utils/util.py +++ b/utils/util.py @@ -253,7 +253,7 @@ def save_frames(image, image_folder, index=None, image_name=None): def folder2vid(image_folder, output_dir, filename): - images = [img for img in os.listdir(image_folder) if img.endswith(".jpg")] + images = [img for img in os.listdir(image_folder) if img.endswith(".jpg") or img.endswith(".png")] images.sort() frame = cv2.imread(os.path.join(image_folder, images[0])) height, width, layers = frame.shape