📜  tensorflow io 检查文件是否存在 - Python (1)

📅  最后修改于: 2023-12-03 15:20:33.859000             🧑  作者: Mango

TensorFlow IO:检查文件是否存在

TensorFlow IO(TensorFlow Input/Output)是一个基于 TensorFlow 的输入输出系统,用于读取和处理各种文件格式。其中包含的一个非常有用的函数就是 tf.io.gfile.exists,它可以用来检查一个文件是否存在。

函数原型
def exists(path):
    """Checks whether a file exists.

    Args:
      path: A string, the path to the file.

    Returns:
      A boolean, whether the file exists or not.
    """
使用方法

使用 tf.io.gfile.exists 函数非常简单,只需将文件路径作为参数传入即可。例如:

import tensorflow as tf

file_path = '/path/to/file.txt'

if tf.io.gfile.exists(file_path):
    print('File exists')
else:
    print('File does not exist')

如果文件存在,则会输出 'File exists';否则输出 'File does not exist'

值得一提的是,tf.io.gfile.exists 函数支持检查本地文件和 GCS(Google Cloud Storage)文件。对于本地文件,函数会直接检查文件系统;而对于 GCS 文件,则需要进行一些前置配置(比如设置 GCS 认证信息等),具体可以参考 官方文档

总结

tf.io.gfile.exists 函数是 TensorFlow IO 库中一个非常实用的函数,能够方便快捷地检查文件是否存在,帮助我们在编写程序时避免出现文件不存在等错误。