from PIL import Image
import os
import glob
def convert_gif_to_jpg(gif_dir, output_dir):
# 创建输出目录(如果不存在)
os.makedirs(output_dir, exist_ok=True)
# 获取目录下所有GIF文件
gif_files = glob.glob(os.path.join(gif_dir, "*.gif"))
for gif_path in gif_files:
try:
with Image.open(gif_path) as gif:
# 提取文件名(不含扩展名)
base_name = os.path.splitext(os.path.basename(gif_path))[0]
frame_num = 0
# 逐帧处理
while True:
try:
gif.seek(frame_num)
frame = gif.copy().convert('RGB')
# 保存为JPG,文件名包含帧号
frame.save(os.path.join(output_dir, f"{base_name}_frame{frame_num}.jpg"), "JPEG")
frame_num += 1
except EOFError:
break # 已处理完所有帧
except Exception as e:
print(f"处理文件 {gif_path} 失败: {e}")
if __name__ == "__main__":
# 使用绝对路径更安全
input_dir = os.path.abspath(r'../data/drive/train/masks/')
output_dir = os.path.abspath(r'../data/drive/train/masks_jpg/')
convert_gif_to_jpg(input_dir, output_dir)
print("转换完成!")
12-20
7103
