Tensorflow从指定链接下载文件

最新推荐文章于 2022-08-16 10:54:40 发布

爱抓猫的狗

最新推荐文章于 2022-08-16 10:54:40 发布

阅读量2.3k

点赞数 1

分类专栏： TensorFlow 文章标签：文件下载

本文链接：https://blog.csdn.net/u012300744/article/details/79784049

版权

TensorFlow 专栏收录该内容

12 篇文章 3 订阅

订阅专栏

在使用Tensorflow的时候，有的时候会遇到从指定连接下载数据集或者其他文件的情况，这里介绍一个使用tf.keras非常方便的函数。

TRAIN_URL = [文件下载链接]
# 返回的是文件保存的路径
def maybe_download():
    train_path = tf.keras.utils.get_file(TRAIN_URL.split('/')[-1], TRAIN_URL)
    # print(train_path)
    test_path = tf.keras.utils.get_file(TEST_URL.split('/')[-1], TEST_URL)
    # print(test_path)
    #返回文件保存的路径，默认在~/.keras文件目录下
    return train_path, test_path