如何在 TensorFlow 中训练图像数据集?

数据挖掘 Python 数据集 张量流 训练 图像识别
2021-10-02 20:04:03

由于我是 TensorFlow 的新手,我想使用 Python 在 TensorFlow 中进行图像识别。对于这个图像识别,我想训练我自己的图像数据集并测试该数据集。请回答我如何训练数据集以及如何选择数据集..

1个回答

如果您对训练用于图像识别的 tensorflow 模型感兴趣,最好阅读 MNIST 上的 tensorflow 教程:

https://www.tensorflow.org/get_started/mnist/beginners https://www.tensorflow.org/get_started/mnist/pros

如果您想自己收集数据并在其上训练模型,这可能会很有趣:https ://digits-draw-recognize.herokuapp.com/

这是我的手写数字识别网站。我自己收集了数据集——我画了大约 1k 位数字并处理了数据。在此之后,我训练了两个模型,其中一个是 Tensorflow 中的 CNN。这些模型在人们绘制的新图像上进行训练。您可以在此站点上找到一些功能描述。