Fix birefnet to 1024x1024
Browse files- utils/add_checkered_background.py +1 -1
- utils/birefnet.py +1 -1
utils/add_checkered_background.py
CHANGED
|
@@ -63,7 +63,7 @@ def process_directory(input_dir, output_dir, square_size=20):
|
|
| 63 |
with ThreadPoolExecutor() as executor:
|
| 64 |
for root, _, files in os.walk(input_dir):
|
| 65 |
for file in files:
|
| 66 |
-
if file.lower().endswith(('.png', '.jpg', '.jpeg')):
|
| 67 |
input_path = os.path.join(root, file)
|
| 68 |
relative_path = os.path.relpath(input_path, input_dir)
|
| 69 |
output_path = os.path.join(output_dir, os.path.splitext(relative_path)[0] + '.png')
|
|
|
|
| 63 |
with ThreadPoolExecutor() as executor:
|
| 64 |
for root, _, files in os.walk(input_dir):
|
| 65 |
for file in files:
|
| 66 |
+
if file.lower().endswith(('.png', '.jpg', '.jpeg', '.webp')):
|
| 67 |
input_path = os.path.join(root, file)
|
| 68 |
relative_path = os.path.relpath(input_path, input_dir)
|
| 69 |
output_path = os.path.join(output_dir, os.path.splitext(relative_path)[0] + '.png')
|
utils/birefnet.py
CHANGED
|
@@ -24,7 +24,7 @@ async def remove_background_birefnet(image_path: str) -> Optional[str]:
|
|
| 24 |
arguments={
|
| 25 |
"image_url": image_path,
|
| 26 |
"model": "General Use (Heavy)",
|
| 27 |
-
"operating_resolution": "
|
| 28 |
"output_format": "png",
|
| 29 |
"refine_foreground": True
|
| 30 |
}
|
|
|
|
| 24 |
arguments={
|
| 25 |
"image_url": image_path,
|
| 26 |
"model": "General Use (Heavy)",
|
| 27 |
+
"operating_resolution": "1024x1024",
|
| 28 |
"output_format": "png",
|
| 29 |
"refine_foreground": True
|
| 30 |
}
|