Skip to main content
 首页 » 编程设计

python之如何使用图像形式的数据和文本文件形式的标签在 tensorflow 中准备我自己的数据集

2025年04月02日32sharpest

我想在 tensorflow 中训练一个卷积神经网络 (CNN),我有 10,000 张 .png 格式的图像数据和 .txt 格式文本文件中的标签,其中每行包含一个标签 (在所有 10,000 行中)。如何在 tensorflow 中准备自己的数据集并训练网络?

请您参考如下方法:

我想您主要关心的是图像部分。要么转换为 jpg 并使用 decode_jpeg , 或使用 ImageFlow转换为 TFRecords。