训练样本中meta_info_GT.txt文件制作（附代码）

一、.txt文件说明

深度学习训练前的训练样本准备中，有时需要meta_info_GT.txt文件，在.txt文件中主要存放图片路径和(h,w,c)信息，见下：

二、.txt文件制作

先在指定文件夹中创建一个xxxxx.txt文件。

使用教程代码需要修改的地方见下：

完整代码见下：

import os
from PIL import Image

def generate_meta_info(folder,output_file):
    # 获取所有序列
    sequences = sorted(os.listdir(folder))

    with open(output_file,"w") as f:
        for seq in sequences:
            seq_folder = os.path.join(folder,seq)
            if not os.path.isdir(seq_folder):
                continue
            # 获取序列中的所有图像
            images = sorted(os.listdir(seq_folder))
            for img_name in images:
                img_path = os.path.join(seq_folder,img_name)
                with Image.open(img_path) as img:
                    width,height = img.size
                    # 记录每个图像的路径和尺寸信息
                    print(img_name)
                    f.write(f"{seq}/{img_name} ({width},{height},3)\n")
                    # f.write(f"{seq}/{img_name} ({width}, {height}, 3)\n")

# 调用函数生成meta_info文件

folder = r"F:\Code\INF_resize_frame_480360\X4"
output_file = r"F:\Code\INF\meta_info_REDS_GT_test.txt"

generate_meta_info(folder,output_file)

最终生成效果如第一小节图片所示。