从链接到tensorflow的tarball中加载图像以训练CNN

时间:2018-10-16 17:40:36

标签: python tensorflow machine-learning download

我是Tensorflow的新手,很难解决一个非常简单的任务。 Tensorflow的大多数示例来自库中内置的MNIST和cifar10,因此我们无需下载* .jpg文件并将其转换为Tensorflow的数组。我有以下数据集:17 flower categories。这是一个包含1360张图像的压缩包。没有图像标签。从第一张图片开始,我们可以将其划分为代表每个类别的80张图片,因此我们获得80x17 = 1360张图片。 我无法将数据集从URL加载到python。但是,我将其下载到本地文件夹,然后使用以下行加载:

import tarfile
f = tarfile.open("../input/17flowers.tgz")
for member in f.getmembers():
   g = f.extractfile(member)

但是,这对于加载图像以训练我的模型没有帮助。 能否请您显示一段代码,从链接加载数据集,然后将图像加载到tensorflow进行训练。每80张连续图像的标签应为1到17。 谢谢!

1 个答案:

答案 0 :(得分:0)

我能够通过引用代码并对其进行修改来解决该问题:https://github.com/changyq12/17_flower_cnn_tensorflow/blob/master/convolutional_network.py