fix config
#2
by
not-lain
- opened
- README.md +6 -7
- config.json +0 -18
README.md
CHANGED
@@ -1,17 +1,16 @@
|
|
1 |
---
|
2 |
library_name: birefnet
|
3 |
tags:
|
4 |
-
-
|
5 |
-
-
|
|
|
|
|
6 |
- Dichotomous Image Segmentation
|
7 |
- Camouflaged Object Detection
|
8 |
- Salient Object Detection
|
9 |
- pytorch_model_hub_mixin
|
10 |
-
- model_hub_mixin
|
11 |
-
repo_url: https://github.com/ZhengPeng7/BiRefNet
|
12 |
-
pipeline_tag: image-segmentation
|
13 |
---
|
14 |
|
15 |
-
This model has been pushed to the Hub using
|
16 |
-
-
|
17 |
- Docs: [More Information Needed]
|
|
|
1 |
---
|
2 |
library_name: birefnet
|
3 |
tags:
|
4 |
+
- model_hub_mixin
|
5 |
+
- Image Segmentation
|
6 |
+
- Background Removal
|
7 |
+
- Mask Generation
|
8 |
- Dichotomous Image Segmentation
|
9 |
- Camouflaged Object Detection
|
10 |
- Salient Object Detection
|
11 |
- pytorch_model_hub_mixin
|
|
|
|
|
|
|
12 |
---
|
13 |
|
14 |
+
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
|
15 |
+
- Library: https://github.com/ZhengPeng7/BiRefNet
|
16 |
- Docs: [More Information Needed]
|
config.json
CHANGED
@@ -1,21 +1,3 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "ZhengPeng7/BiRefNet",
|
3 |
-
"architectures": [
|
4 |
-
"BiRefNet"
|
5 |
-
],
|
6 |
-
"auto_map": {
|
7 |
-
"AutoConfig": "BiRefNet_config.BiRefNetConfig",
|
8 |
-
"AutoModelForImageSegmentation": "birefnet.BiRefNet"
|
9 |
-
},
|
10 |
-
"custom_pipelines": {
|
11 |
-
"image-segmentation": {
|
12 |
-
"impl": "BiRefNet_pipe.BiRefNetPipe",
|
13 |
-
"pt": [
|
14 |
-
"AutoModelForImageSegmentation"
|
15 |
-
],
|
16 |
-
"tf": [],
|
17 |
-
"type": "image"
|
18 |
-
}
|
19 |
-
},
|
20 |
"bb_pretrained": false
|
21 |
}
|
|
|
1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
"bb_pretrained": false
|
3 |
}
|