chinese-support

MyPipe.py

@@ -50,9 +50,9 @@ class RMBGPipe(Pipeline):
     pil_im = Image.fromarray(result_image)
     if return_mask ==True : 
       return pil_im
+    no_bg_image = Image.new("RGBA", pil_im.size, (0,0,0,0))
     input_image = load_image(input_image)
-    no_bg_image = input_image.copy()
-    no_bg_image.putalpha(pil_im)
+    no_bg_image.paste(input_image, mask=pil_im)
     return no_bg_image
     
   # utilities functions

README.md

@@ -11,7 +11,6 @@ tags:
 - vision
 - legal liability
 - transformers
-- transformers.js
 
 extra_gated_description: RMBG v1.4 is available as a source-available model for non-commercial use
 extra_gated_heading: "Fill in this form to get instant access"
@@ -35,27 +34,16 @@ It is ideal where content safety, legally licensed datasets, and bias mitigation
 
 Developed by BRIA AI, RMBG v1.4 is available as a source-available model for non-commercial use. 
 
-
-To purchase a commercial license, simply click [Here](https://go.bria.ai/3D5EGp0).
-
-
 [CLICK HERE FOR A DEMO](https://huggingface.co/spaces/briaai/BRIA-RMBG-1.4)
-
-**NOTE** New RMBG version available! Check out [RMBG-2.0](https://huggingface.co/briaai/RMBG-2.0)
-
-Join our [Discord community](https://discord.gg/Nxe9YW9zHS) for more information, tutorials, tools, and to connect with other users!
-
-
 ![examples](t4.png)
 
-
 ### Model Description
 
 - **Developed by:** [BRIA AI](https://bria.ai/)
 - **Model type:** Background Removal 
 - **License:** [bria-rmbg-1.4](https://bria.ai/bria-huggingface-model-license-agreement/)
   - The model is released under a Creative Commons license for non-commercial use.
-  - Commercial use is subject to a commercial agreement with BRIA. To purchase a commercial license simply click [Here](https://go.bria.ai/3B4Asxv).
+  - Commercial use is subject to a commercial agreement with BRIA. [Contact Us](https://bria.ai/contact-us) for more information. 
 
 - **Model Description:** BRIA RMBG 1.4 is a saliency segmentation model trained exclusively on a professional-grade dataset.
 - **BRIA:** Resources for more information: [BRIA AI](https://bria.ai/)
@@ -124,10 +112,6 @@ pillow_image = pipe(image_path) # applies mask on input and returns a pillow ima
 
 Or load the model 
 ```python
-from PIL import Image
-from skimage import io
-import torch
-import torch.nn.functional as F
 from transformers import AutoModelForImageSegmentation
 from torchvision.transforms.functional import normalize
 model = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4",trust_remote_code=True)
@@ -157,7 +141,6 @@ model.to(device)
 image_path = "https://farm5.staticflickr.com/4007/4322154488_997e69e4cf_z.jpg"
 orig_im = io.imread(image_path)
 orig_im_size = orig_im.shape[0:2]
-model_input_size = [1024, 1024]
 image = preprocess_image(orig_im, model_input_size).to(device)
 
 # inference 
@@ -167,9 +150,9 @@ result=model(image)
 result_image = postprocess_image(result[0][0], orig_im_size)
 
 # save result
-pil_mask_im = Image.fromarray(result_image)
+pil_im = Image.fromarray(result_image)
+no_bg_image = Image.new("RGBA", pil_im.size, (0,0,0,0))
 orig_image = Image.open(image_path)
-no_bg_image = orig_image.copy()
-no_bg_image.putalpha(pil_mask_im)
+no_bg_image.paste(orig_image, mask=pil_im)
 ```
 

example_inference.py

@@ -28,10 +28,10 @@ def example_inference():
     result_image = postprocess_image(result[0][0], orig_im_size)
 
     # save result
-    pil_mask_im = Image.fromarray(result_image)
+    pil_im = Image.fromarray(result_image)
+    no_bg_image = Image.new("RGBA", pil_im.size, (0,0,0,0))
     orig_image = Image.open(im_path)
-    no_bg_image = orig_image.copy()
-    no_bg_image.putalpha(pil_mask_im)    
+    no_bg_image.paste(orig_image, mask=pil_im)
     no_bg_image.save("example_image_no_bg.png")