【Openmv使用笔记】
在采集用户信息入库和人脸识别遍历库内文件的时候,需要读取目录下文件数,参考网上程序
#统计文件夹下文件的数目
import os
DIR = '/tmp' #要统计的文件夹
print len([name for name in os.listdir(DIR) if os.path.isfile(os.path.join(DIR, name))])
#如统计文件夹数量,用 os.path.isdir(path)做判断语句。
报错找不到isdir或者isfile,原因在于micropython中没有这两个函数
鉴于库中子目录只存放文件夹,所以后来采用了简单的
len(os.listdir(PATH))
即可
优化
上述方法存在一定缺陷。如果库中删去了其中某个子目录(某位用户的信息),则数字不连续,识别过程中会读取不到发生错误,所以采用一下方法获得子目录个数
def dirnum(p):
dirlist = os.listdir(p)
dirlist.sort()#排序,实验过程中发生了子目录不按序号排序的情况
sn = dirlist[-1]#目标为列表的最后一位,即子目录中序号最大的目录
n = int(sn[-2:])#取最后两位并转化成整数
return n
在读取时,如下所示加上一个判断
...
for s in range(1, N_MEMBER+1):
sn = "s{0:0>2}".format(s)
if(sn in dirlist):
...
可滤掉不存在的子目录