1580158
/
sd-webui-segment-anything
mirror of https://github.com/continue-revolution/sd-webui-segment-anything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: add query available sam model endpoint

pull/90/head
chace20 2023-05-10 15:52:15 +08:00
parent a4b249114e
commit 1a82dde6f0
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/api.py
View File

@ -8,6 +8,7 @@ import numpy as np
from modules.api.api import encode_pil_to_base64, decode_base64_to_image from modules.api.api import encode_pil_to_base64, decode_base64_to_image
from scripts.sam import sam_predict, dino_predict, update_mask, cnet_seg, categorical_mask from scripts.sam import sam_predict, dino_predict, update_mask, cnet_seg, categorical_mask
from scripts.sam import sam_model_list
def decode_to_pil(image): def decode_to_pil(image):
@ -42,6 +43,10 @@ def sam_api(_: gr.Blocks, app: FastAPI):
"msg": "Success!" "msg": "Success!"
} }
@app.get("/sam/sam-model", description='Query available SAM model')
async def api_sam_model() -> List[str]:
return sam_model_list
class SamPredictRequest(BaseModel): class SamPredictRequest(BaseModel):
sam_model_name: str = "sam_vit_h_4b8939.pth" sam_model_name: str = "sam_vit_h_4b8939.pth"
input_image: str input_image: str