我想训练张量流中的卷积神经网络(CNN),我有10,000个.png格式的图像数据和一个.txt格式的文本文件中的标签,其中每行包含一个标签(全部10,000个行)。如何在张量流中准备自己的数据集并训练网络?
最佳答案
我想您主要关心的是图像部分。要么转换为jpg并使用decode_jpeg,要么使用ImageFlow转换为TFRecords。
我想训练张量流中的卷积神经网络(CNN),我有10,000个.png格式的图像数据和一个.txt格式的文本文件中的标签,其中每行包含一个标签(全部10,000个行)。如何在张量流中准备自己的数据集并训练网络?
最佳答案
我想您主要关心的是图像部分。要么转换为jpg并使用decode_jpeg,要么使用ImageFlow转换为TFRecords。