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

update default sam_prompt structure to include point data

Browse Source 
Signed-off-by: ふぁ <yuki@yuki0311.com>
This commit is contained in:
ふぁ 2024-11-29 13:21:10 +09:00
parent 00ad95d1c0
commit 9e6c46184d
No known key found for this signature in database
GPG Key ID: 83A8A5E74872A8AA
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
rembg/sessions/sam.py
View File

@ -143,7 +143,16 @@ class SamSession(BaseSession):
Returns: Returns:
List[PILImage]: A list of masks generated by the decoder. List[PILImage]: A list of masks generated by the decoder.
""" """
prompt = kwargs.get("sam_prompt", "{}") prompt = kwargs.get(
"sam_prompt",
[
{
"type": "point",
"label": 1,
"data": [int(img.width / 2), int(img.height / 2)],
}
],
)
schema = { schema = {
"type": "array", "type": "array",
"items": { "items": {