From d188ef2b336936d16f633ff9ec21afb372901b9e Mon Sep 17 00:00:00 2001 From: caiwx86 Date: Tue, 6 Dec 2022 17:18:33 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=9B=BE=E7=89=87=E6=8E=92?= =?UTF-8?q?=E5=BA=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/ImageUtils.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/utils/ImageUtils.py b/utils/ImageUtils.py index 2868de8..a4978c0 100644 --- a/utils/ImageUtils.py +++ b/utils/ImageUtils.py @@ -1,4 +1,5 @@ import base64,hashlib,os,shutil +import numpy as np from PIL import Image class imageUtils: @@ -91,8 +92,20 @@ class imageUtils: #image_height = 376 #image_width = 720 images_format = ['.png','.jpg'] - image_names = [name for name in os.listdir(src) for item in images_format if - os.path.splitext(name)[1] == item][::-1] + + #image_names = [name for name in os.listdir(src) for item in images_format if + # os.path.splitext(name)[1] == item][::-1] + img_list=os.listdir(src) + img_list.sort() + img_list.sort(key=lambda x: int(x[:-4])) + ##文件名按数字排序 + img_nums=len(img_list) + image_names = [] + for i in range(img_nums): + img_name=src+img_list[i] + image_names.append(img_name) + #使用倒序 + image_names = image_names[::-1] # 简单的对于参数的设定和实际图片集的大小进行数量判断 if len(image_names) < row * column: raise ValueError("合成图片的参数和要求的数量不能匹配！")