在TensorFlow中,可以使用tf.data.Dataset API來處理多張圖片。以下是一種常見的處理方式:
import glob
image_files = glob.glob('path_to_images/*.jpg')
dataset = tf.data.Dataset.from_tensor_slices(image_files)
def preprocess_image(image_file):
image = tf.io.read_file(image_file)
image = tf.image.decode_jpeg(image, channels=3)
image = tf.image.resize(image, [224, 224])
image = image / 255.0 # 歸一化到 [0, 1] 范圍
return image
dataset = dataset.map(preprocess_image)
def augment_image(image):
image = tf.image.random_crop(image, [200, 200, 3])
image = tf.image.random_flip_left_right(image)
return image
dataset = dataset.map(augment_image)
dataset = dataset.shuffle(1000)
dataset = dataset.batch(32)
for images in dataset:
# 進行模型訓練或者預測
...
通過以上步驟,就可以使用TensorFlow處理多張圖片數據了。根據具體的需求,可以根據實際情況調整預處理和數據增強的操作。