python通过os.walk读取文件夹中的连续图片

最新推荐文章于 2023-07-14 14:36:11 发布

原创

最新推荐文章于 2023-07-14 14:36:11 发布 · 2.4k 阅读

2 ·

CC 4.0 BY-SA版权

本文介绍如何使用Python的os.walk遍历文件夹，并按时间戳排序处理连续图片，无需预先保存图片名称到txt文件。

之前读取并处理文件夹中的连续图片，都是先将其时间戳或者文件名保存到txt。处理时读取txt文件。

今天尝试了python中的os.walk。不需要事先保存图像名称，很方便。

代码如下：

#!/usr/bin/env python
print "importing libraries"
import time, sys, os
import argparse
import cv2
import numpy as np
import csv

#structure
# dataset/cam0/TIMESTAMP.png
# dataset/camN/TIMESTAMP.png
# dataset/imu.csv

#input commid 
#python readPicture.py --folder dataset
#参考http://www.runoob.com/python/os-walk.html

#setup the argument list
parser = argparse.ArgumentParser(description='Extract a ROS bag containing a image and imu topics.')
parser.add_argument('--folder',  metavar='folder', nargs='?', help='Data folder')

#print help if no argument is specified
if len(sys.argv)<1:
    parser.print_help()
    sys.exit(0)

#parse the args
parsed = parser.parse_args()

def getImageFilesFromDir(dir):
    '''Generates a list of files from the directory'''
    image_files = list()
    timestamps = list()
    if os.path.exists(dir):
        for p