如何对数据集的文件名进行排序，我使用 glob.glob('path/*.png')

Question

图片

你好。 我正在尝试加载数据集，我想将文件名排序为 1、2、3 ...，但我将它们作为 1、10、100、1000 ....，正如您在图像中看到的那样？ 我应该如何完成它？

filenames = glob.glob("/content/path/train/*.png")
filenames.sort()
images = [cv2.imread(img) for img in filenames]

Answer 1

嗯，这是字母顺序...如果您知道名称都是数字，那么您可以提供一个key function 到.sort()使用您想要的顺序。

def bynumber(fn):
    return int(os.path.basename(fn)[:-4])
...
filenames.sort( key=bynumber )

或者：

filenames.sort( key=lambda fn: int(os.path.basename(fn)[:-4]) )