使用TensorFlow来读取一个文件夹中所有bin文件,并转换成3通道,固定大小的矩阵步骤:
TensorFlow版本:1.15.0
- 导入文件夹下所有文件名:
import os
dir_path = "./文件夹路径"
filenames = os.listdir(dir_path)
file_list = [os.path.join(dir_path, file) for file in filenames]
- 根据文件名列表读取文件:
import tensorflow as tf
file_queue = tf.train.string_input_producer(file_list)
#读取阶段
reader = tf.FixedLengthRecordReader(self.all_bytes)