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

Update remove_bg.py

Browse Source 
only remove non png images
This commit is contained in:
Emil Mirzayev 2024-10-12 20:29:06 +01:00 committed by GitHub
parent a222f3e14b
commit 03f31859f1
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: with tqdm(total=total_files, desc="Processing images", unit="image") as pbar:
for subdir, dirs, files in os.walk(directory): for subdir, dirs, files in os.walk(directory):
for file in files: for file in files:
if file.lower().endswith(('.jpg', '.jpeg', 'png')): if file.lower().endswith(('.jpg', '.jpeg')):
input_path = os.path.join(subdir, file) input_path = os.path.join(subdir, file)
output_filename = os.path.splitext(file)[0] + '.png' output_filename = os.path.splitext(file)[0] + '.png'
output_path = os.path.join(subdir, output_filename) output_path = os.path.join(subdir, output_filename)