如何用Python批量去除多张图片背景？代码报错求助

原代码存在的问题 #

• 语法错误：for循环后的代码未缩进，Python依赖缩进识别代码块，这是第6行报错的核心原因
• 路径拼接错误：文件夹路径与文件名直接拼接，缺少路径分隔符，生成的路径无效
• 输出路径错误：指定的输出路径是文件夹而非具体文件，无法直接保存图片
• 未过滤非图片文件：遍历会包含所有文件，可能尝试打开非图片文件引发错误

修正后的代码 #

from rembg import remove
from PIL import Image
import os

# 定义输入输出文件夹路径
input_folder = r'C:\Users\Carlo Karim\Desktop\folder'
output_folder = r'C:\Users\Carlo Karim\Documents\output'

# 确保输出文件夹存在，不存在则自动创建
os.makedirs(output_folder, exist_ok=True)

# 遍历输入文件夹中的文件
for filename in os.listdir(input_folder):
    # 仅处理常见图片格式
    if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
        # 安全拼接输入输出文件路径
        input_path = os.path.join(input_folder, filename)
        output_path = os.path.join(output_folder, filename)
        
        try:
            # 打开图片并去除背景
            input_img = Image.open(input_path)
            output_img = remove(input_img)
            # 保存处理后的图片
            output_img.save(output_path)
            print(f"处理完成：{filename}")
        except Exception as e:
            print(f"处理{filename}失败：{str(e)}")

关键修改说明 #

• 修复缩进：统一for循环内代码的缩进，符合Python语法要求
• 用os.path.join拼接路径：自动适配系统路径分隔符，避免手动拼接的错误
• 预创建输出文件夹：防止因文件夹不存在导致保存失败
• 过滤图片文件：只处理指定格式的图片，避免无效文件引发错误
• 添加异常处理：单个文件处理出错时不会中断整个程序，同时输出错误信息
• 变量重命名：将input改为input_img，避免与Python内置函数冲突

内容的提问来源于stack exchange，提问作者Carlo Karim