AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-05 18:50:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix typos in USAGE.md

Browse Source
This commit is contained in:
Succubyss 2023-04-21 15:27:45 -05:00 committed by GitHub
parent 91379e8d78
commit c3fd1b430e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
USAGE.md
View File

@ -52,15 +52,15 @@ output = remove(input, bgcolor=(255, 255, 255))
```
### Using input points
You can use the `input_point` and `input_label` argument to specify the points that should be used for the masks. This only works with the `sam` model.
You can use the `input_points` and `input_labels` arguments to specify the points that should be used for the masks. This only works with the `sam` model.
```python
import numpy as np
# Define the points and labels
# The points are defined as [y, x]
input_point = np.array([[400, 350], [700, 400], [200, 400]])
input_label = np.array([1, 1, 2])
input_points = np.array([[400, 350], [700, 400], [200, 400]])
input_labels = np.array([1, 1, 2])
image = remove(image,session=session, input_point=input_point, input_label=input_label)
image = remove(image,session=session, input_points=input_points, input_labels=input_labels)
```
## Save the image