Karl/comfy_fm24_newgens
1
0
Fork 0
mirror of https://github.com/karl0ss/comfy_fm24_newgens.git synced 2025-04-28 20:03:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update remove_bg.py

Browse Source
This commit is contained in:
Emil Mirzayev 2024-10-12 20:30:52 +01:00 committed by GitHub
parent a8ed7c5821
commit 0eec96b787
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
remove_bg.py
View File

@ -45,7 +45,7 @@ def process_directory(directory):
with tqdm(total=total_files, desc="Processing images", unit="image") as pbar:
for subdir, dirs, files in os.walk(directory):
for file in files:
if file.lower().endswith(('.jpg', '.jpeg')):
if file.lower().endswith(('.jpg', '.jpeg', 'png')):
input_path = os.path.join(subdir, file)
output_filename = os.path.splitext(file)[0] + '.png'
output_path = os.path.join(subdir, output_filename)