使用 python 从以数字名称标签索引的文件夹中读取图像文件

Question

我正在尝试使用 python 从文件夹中读取一系列图像。这些图像表示已拆分为网格的小块较大图像，并使用“imagename_row_column.jpg”（即 img_0_1.jpg）进行索引。我当前的代码（粘贴在下面）在列索引方面有问题，并且以错误的顺序计算 10 及以上的数字。例如，而不是阅读像 (img_0, img_0_1, img_0_2,...img_0_9, img_0_10...) 我是getting (img_0, img_0_10, img_0_11, img_0_1, img_0_2...) 任何建议将不胜感激。谢谢！

# Get images from folder
path1 = r'C:\Users\user_\Desktop\Test\IMG_Scan'
images = []
mylist = os.listdir(path1)
for img in mylist:
    curimg = cv2.imread(f'{path1}/{img}')
    images.append(curimg)

Answer 1

"img_0_10" < "img_0_2" 通过字符串比较。在迭代之前尝试对文件进行自定义排序：

mylist = sorted(os.listdir(path1), key=lambda x: list(map(int, x.split("_")[1:])))

使用 python 从以数字名称标签索引的文件夹中读取图像文件

Reading in image files from a folder indexed with numeric name tags using python

python

path