Karl/comfy_fm24_newgens
1
0
Tükrözés 0
tükörképe a: https://github.com/karl0ss/comfy_fm24_newgens.git synced 2025-11-24 12:04:30 +00:00
Kód Hibajegyek Packages Projektek Kiadások Wiki Tevékenység

Update remove_bg.py

Forráskód böngészése
This commit is contained in:
Emil Mirzayev 2024-10-12 20:30:52 +01:00 committed by GitHub
szülő a8ed7c5821
commit 0eec96b787
Nem található kulcs ehhez az aláíráshoz az adatbázisban
GPG kulcs azonosító: B5690EEEBB952194
1 fájl változott, egészen pontosan 1 új sor hozzáadva és 1 régi sor törölve
Statisztikák mutatása Download Patch File Download Diff File Expand all files Collapse all files

2
remove_bg.py
Fájl megtekintése

@ -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')): if file.lower().endswith(('.jpg', '.jpeg', 'png')):
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)