之前读取并处理文件夹中的连续图片,都是先将其时间戳或者文件名保存到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