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("合成图片的参数和要求的数量不能匹配!")