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