tensorflow list_files需要注意的点

tensorflow 读取tfrecord的时候 list_files几乎是避不开的函数,在tensorflow的官方文档中写了,list_files 支持正则表达式。

但是这里面有个比较坑的点:我们点开list_files的源码,会发现调用了gen_io_ops.matching_files。这个是tf.io.matching_files函数的底层算子,我们打开这个函数会发现一个注意事项

Note that this routine only supports wildcard characters in the basename portion of the pattern, not in the directory portion. Note also that the order of filenames returned is deterministic.

即该函数仅支持最后一层文件级别的正则表达式,而不支持目录的正则表达式。

不过这也不代表我们没办法读多个目录,list_files的入参可以是一个路径字符串,也可以是一个list,其中的每个元素都可以是一个路径正则表达式

上一篇:会议室预约小程序开源版开发


下一篇:VMWare ubuntu 18.04 网卡丢失