add custom architecture
Browse files- README.md +2 -2
- config.json +25 -0
- model.safetensors +3 -0
README.md
CHANGED
@@ -30,7 +30,7 @@ general stock images, e-commerce, gaming, and advertising content, making it sui
|
|
30 |
The accuracy, efficiency, and versatility currently rival leading source-available models.
|
31 |
It is ideal where content safety, legally licensed datasets, and bias mitigation are paramount.
|
32 |
|
33 |
-
Developed by BRIA AI, RMBG v1.4 is available as
|
34 |
|
35 |
[CLICK HERE FOR A DEMO](https://huggingface.co/spaces/briaai/BRIA-RMBG-1.4)
|
36 |
![examples](t4.png)
|
@@ -40,7 +40,7 @@ Developed by BRIA AI, RMBG v1.4 is available as a source-available model for non
|
|
40 |
- **Developed by:** [BRIA AI](https://bria.ai/)
|
41 |
- **Model type:** Background Removal
|
42 |
- **License:** [bria-rmbg-1.4](https://bria.ai/bria-huggingface-model-license-agreement/)
|
43 |
-
- The model is released under
|
44 |
- Commercial use is subject to a commercial agreement with BRIA. [Contact Us](https://bria.ai/contact-us) for more information.
|
45 |
|
46 |
- **Model Description:** BRIA RMBG 1.4 is a saliency segmentation model trained exclusively on a professional-grade dataset.
|
|
|
30 |
The accuracy, efficiency, and versatility currently rival leading source-available models.
|
31 |
It is ideal where content safety, legally licensed datasets, and bias mitigation are paramount.
|
32 |
|
33 |
+
Developed by BRIA AI, RMBG v1.4 is available as an source-available model for non-commercial use.
|
34 |
|
35 |
[CLICK HERE FOR A DEMO](https://huggingface.co/spaces/briaai/BRIA-RMBG-1.4)
|
36 |
![examples](t4.png)
|
|
|
40 |
- **Developed by:** [BRIA AI](https://bria.ai/)
|
41 |
- **Model type:** Background Removal
|
42 |
- **License:** [bria-rmbg-1.4](https://bria.ai/bria-huggingface-model-license-agreement/)
|
43 |
+
- The model is released under an Creative Commons license for non-commercial use.
|
44 |
- Commercial use is subject to a commercial agreement with BRIA. [Contact Us](https://bria.ai/contact-us) for more information.
|
45 |
|
46 |
- **Model Description:** BRIA RMBG 1.4 is a saliency segmentation model trained exclusively on a professional-grade dataset.
|
config.json
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "not-lain/CustomCodeForRMBG",
|
3 |
+
"architectures": [
|
4 |
+
"BriaRMBG"
|
5 |
+
],
|
6 |
+
"auto_map": {
|
7 |
+
"AutoConfig": "not-lain/CustomCodeForRMBG--MyConfig.RMBGConfig",
|
8 |
+
"AutoModelForImageSegmentation": "not-lain/CustomCodeForRMBG--briarmbg.BriaRMBG"
|
9 |
+
},
|
10 |
+
"custom_pipelines": {
|
11 |
+
"image-segmentation": {
|
12 |
+
"impl": "not-lain/CustomCodeForRMBG--MyPipe.RMBGPipe",
|
13 |
+
"pt": [
|
14 |
+
"AutoModelForImageSegmentation"
|
15 |
+
],
|
16 |
+
"tf": [],
|
17 |
+
"type": "image"
|
18 |
+
}
|
19 |
+
},
|
20 |
+
"in_ch": 3,
|
21 |
+
"model_type": "SegformerForSemanticSegmentation",
|
22 |
+
"out_ch": 1,
|
23 |
+
"torch_dtype": "float32",
|
24 |
+
"transformers_version": "4.38.0.dev0"
|
25 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:46ef7fe46f2ae284d8f1aaa24bfa5fca5ef25a34e2c7caa890a0029eb100e87f
|
3 |
+
size 176381984
|