如何使用图像和标签制作数据集？

Question

*初学者

我正在建立一个神经网络，以确定汽车是模型a还是模型b。基于此预测，我希望网络告诉我汽车的功率。我的问题是，如何创建具有图像，模型标签和功率的数据集。然后如何将其放入numpy数组中。

Answer 1

我建议您将图像及其标签（模型a或b）及其功能（马力）存储在TFRecord中，该文件可让您将数据库保持在Tensorflow随时可用的状态。

关于这个主题，这里有一个很好的教程：https://www.tensorflow.org/tutorials/load_data/tfrecord

this stackoverflow answer将为您提供有关如何从TFRecord文件提取数据的其他帮助。