一、.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)
最终生成效果如第一小节图片所示。