python对文件的操作

读取指定文件夹中的图片名称和数量

seq_info = "F:\\demo\\car" #指定文件夹

image_filenames = {int(os.path.splitext(f)[0]): os.path.join(seq_info, f) 
    for f in os.listdir(seq_info)}  #获取seq_info文件夹中的jpg图片名称
print(len(image_filenames)) #输出图片数量

获取文件夹中的所有图片名：

data_dir = 'F:\\data\\vedio\\bicycle'
img_file = os.listdir(data_dir)
img_ = sorted([file for file in img_file if file.endswith('jpg')])

for i in range(len(img_)):
    img_list = os.path.join(data_dir,img_[i])
    print(img_list)  #输出图片完整路径名

读取txt文件中的坐标：np.genfromtxt（）函数，genfromtxt函数创建数组表格数据。主要执行两个循环运算。第一个循环将文件的每一行转换成字符串序列。第二个循环将每个字符串序列转换为相应的数据类型。

with open(self.root_dir  + folder + '/groundtruth.txt') as f:
    gtbox = np.genfromtxt(f, delimiter=',')  # 读取txt中的数据

读到的直接就是转换好的类型了，十分方便，