如何制作带有图像和标签的数据集？

Question

*初学者

我正在构建一个神经网络，用于确定汽车是 a 型还是 b 型。根据这个预测，我希望网络告诉我汽车的马力。我的问题是，如何创建包含图像、模型标签和马力的数据集。然后我如何把它放在一个 numpy 数组中。

Answer 1

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

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

this stackoverflow answer 将在如何从 TFRecord 文件中提取数据方面为您提供更多帮助。