import pandas as pd
import os
def getfile(photo_path):
filelist = os.listdir(photo_path)
num_jpg = len(filelist)
print("该文件夹下一共有{}张".format(num_jpg))
#--------------排序-------------
allTestDataName = [] #不需要排序可以注释
for filename in filelist:
if filename.endswith('.jpg'):
allTestDataName.append(filename)
allTestDataName.sort(key= lambda x:int(x[:-4])) #做一个排序 前提是有循序的图片如:001.jpg,002.jpg,003.jpg等等
#--------------排序-------------
with open("MOT20.txt","a") as f:
for item in allTestDataName: #不需要排序把allTestDataName改成filelist
item=photo_path+item
f.write(item)
f.write('\n')
def main():
list_name=["MOT20/images/train/MOT20-01/img1/","MOT20/images/train/MOT20-02/img1/"] #一定要以“/”结尾
for i in list_name:
getfile(i)
if __name__ == '__main__':
main()
将文件夹的所有图片路径用pandas写入txt内,可排序
最新推荐文章于 2022-04-30 15:29:20 发布