首页tensorflow python 读取文件夹图片，训练，预测，保存结果

tensorflow python 读取文件夹图片，训练，预测，保存结果

时间: 2023-05-20 19:07:05 浏览: 133

tensorflow实现对图片的读取的示例代码

tensorflow里面给出了一个函数用来读取图像，不过得到的结果是最原始的图像，是咩有经过解码的图像，这个函数为tf.gfile.FastGFile（‘path’， ‘r’）.read()。如果要显示读入的图像，那就需要经过解码过程，tensorflow里面提供解码的函数有两个，tf.image.decode_jepg和tf.image.decode_png分别用于解码jpg格式和png格式的图像进行解码，得到图像的像素值，这个像素值可以用于显示图像。如果乜有解码，读取的图像是一个字符串，没法显示。例如： import matplotlib.pyplot as plt; import t

您可以使用TensorFlow中的tf.data.Dataset API来读取文件夹中的图片，进行训练和预测。您可以使用tf.keras模块来构建模型，并使用fit()函数进行训练。在预测时，您可以使用predict()函数来获取预测结果。最后，您可以使用save()函数将模型保存到磁盘上。

阅读全文