如何使用Tensorflow将花卉数据集下载到环境中?


花卉数据集可以使用一个Google API下载,该API基本上链接到花卉数据集。“get_file”方法可以用来传递API作为参数。完成此操作后,数据将下载到环境中。

阅读更多: 什么是TensorFlow以及Keras如何与TensorFlow一起创建神经网络?

我们将使用花卉数据集,其中包含数千朵花的图像。它包含5个子目录,每个类别都有一个子目录。

我们使用Google Colaboratory来运行以下代码。Google Colab或Colaboratory帮助通过浏览器运行Python代码,无需任何配置,并且可以免费访问GPU(图形处理单元)。Colaboratory构建在Jupyter Notebook之上。

import numpy as np
import os
import PIL
import PIL.Image
import tensorflow as tf
import tensorflow_datasets as tfds
print("The Tensorflow version is :")
print(tf.__version__)

import pathlib
print("The dataset is being downloaded")
dataset_url = "https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz"
data_dir = tf.keras.utils.get_file(origin=dataset_url,
fname='flower_photos',
untar=True)
data_dir = pathlib.Path(data_dir)

print("The number of images in the dataset")
image_count = len(list(data_dir.glob('*/*.jpg')))
print(image_count)

print("Sample data in the dataset")
roses = list(data_dir.glob('roses/*'))
PIL.Image.open(str(roses[2]))

print("Some more sample data from the dataset")
roses = list(data_dir.glob('roses/*'))
PIL.Image.open(str(roses[4]))

代码来源: https://tensorflowcn.cn/tutorials/load_data/images

输出

The Tensorflow version is :
2.4.0
The dataset is being downloaded
Downloading data from https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz
228818944/228813984 [==============================] - 4s 0us/step
The number of images in the dataset
3670
Sample data in the dataset
Some more sample data from the dataset

解释

  • 下载所需软件包。
  • 在控制台上显示Tensorflow的版本。
  • 使用API下载数据。
  • 在控制台上还显示图像示例。

更新于: 2021年2月19日

341 次浏览

开启您的 职业生涯

通过完成课程获得认证

开始学习
广告