README.md: tweaks
Browse files
README.md
CHANGED
@@ -58,7 +58,7 @@ It's a prerequisite for a number of object manipulation tasks like:
|
|
58 |
- Replace an object in an image
|
59 |
- ...
|
60 |
|
61 |
-
Out-of-the box, the Finegrain
|
62 |
|
63 |
## Training
|
64 |
|
@@ -85,7 +85,7 @@ Given our focus on e-commerce, we crafted a specific test set, and in order to e
|
|
85 |
|
86 |
We're using the usual metrics, namely MAE, Smeasure, Emeasure and Dice, computed with [PySODMetrics](https://github.com/lartpang/PySODMetrics). We'll add more later to account for matting aspects (transparent objects) - still a work-in-progress on our end.
|
87 |
|
88 |
-
| Model | **MAE** β | **Smeasure** β | **
|
89 |
|--|--|--|--|--|
|
90 |
| `briaai/RMBG-1.4` (x) | 0.0226 | 90.7% | 94.3% | 88.5% |
|
91 |
| `ZhengPeng7/BiRefNet` (xx) | 0.0194 | 93.1% | 95.1% | 91.5% |
|
@@ -172,7 +172,7 @@ mask = segmenter(input_image, box_prompt=(24, 133, 588, 531))
|
|
172 |
mask.save("output.png")
|
173 |
```
|
174 |
|
175 |
-
### With
|
176 |
|
177 |
Install the `comfyui-refiners` custom nodes:
|
178 |
```
|
|
|
58 |
- Replace an object in an image
|
59 |
- ...
|
60 |
|
61 |
+
Out-of-the box, the Finegrain Box Segmenter requires a bounding box as an input, but you can easily augment it to enable "prompt to select object" scenarios - see the [Finegrain Object Cutter Hugging Face space](https://huggingface.co/spaces/finegrain/finegrain-object-cutter) for an example implementation.
|
62 |
|
63 |
## Training
|
64 |
|
|
|
85 |
|
86 |
We're using the usual metrics, namely MAE, Smeasure, Emeasure and Dice, computed with [PySODMetrics](https://github.com/lartpang/PySODMetrics). We'll add more later to account for matting aspects (transparent objects) - still a work-in-progress on our end.
|
87 |
|
88 |
+
| Model | **MAE** β | **Smeasure** β | **Emeasure** β | **Dice** β |
|
89 |
|--|--|--|--|--|
|
90 |
| `briaai/RMBG-1.4` (x) | 0.0226 | 90.7% | 94.3% | 88.5% |
|
91 |
| `ZhengPeng7/BiRefNet` (xx) | 0.0194 | 93.1% | 95.1% | 91.5% |
|
|
|
172 |
mask.save("output.png")
|
173 |
```
|
174 |
|
175 |
+
### With ComfyUI (https://registry.comfy.org/publishers/finegrain/nodes/comfyui-refiners)
|
176 |
|
177 |
Install the `comfyui-refiners` custom nodes:
|
178 |
```
|