Skip to content

Instantly share code, notes, and snippets.

@ChrisTitusTech

ChrisTitusTech/removebg.py

Created February 3, 2024 19:40
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ChrisTitusTech/3043408c112a17d06f966c7bf4cc2c32 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ChrisTitusTech/3043408c112a17d06f966c7bf4cc2c32 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. ChrisTitusTech created this gist Feb 3, 2024.
    20 changes: 20 additions & 0 deletions removebg.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,20 @@
    import os
    import rembg

    def process_images(input_dir, output_dir):
    if not os.path.exists(output_dir):
    os.makedirs(output_dir)

    for filename in os.listdir(input_dir):
    if filename.lower().endswith('.png'):
    input_path = os.path.join(input_dir, filename)
    output_path = os.path.join(output_dir, f"{os.path.splitext(filename)[0]}_output.png")

    with open(input_path, 'rb') as input_file, open(output_path, 'wb') as output_file:
    output_file.write(rembg.remove(input_file.read()))

    if __name__ == "__main__":
    input_directory = os.getcwd() # current directory
    output_directory = os.path.join(input_directory, 'alpha')

    process_images(input_directory, output_directory)