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 | 
             
                        }
         |