修复图片排序

This commit is contained in:
caiwx86 2022-12-06 17:18:33 +08:00
parent 136fe64328
commit d188ef2b33

View File

@ -1,4 +1,5 @@
import base64,hashlib,os,shutil import base64,hashlib,os,shutil
import numpy as np
from PIL import Image from PIL import Image
class imageUtils: class imageUtils:
@ -91,8 +92,20 @@ class imageUtils:
#image_height = 376 #image_height = 376
#image_width = 720 #image_width = 720
images_format = ['.png','.jpg'] 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: if len(image_names) < row * column:
raise ValueError("合成图片的参数和要求的数量不能匹配!") raise ValueError("合成图片的参数和要求的数量不能匹配!")