aidlab/aida_seg_anything
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
zcr
2026-03-27 14:47:25 +08:00
parent dd3c070f4d
commit eda2acd1a4
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@@ -1,17 +1,14 @@
import io
import os import os
import urllib.request # 必须这样写，不能只 import urllib import urllib.request
import uuid
from typing import Optional, List from typing import Optional, List
import cv2 import cv2
import litserve as ls import litserve as ls
import numpy as np import numpy as np
import torch import torch
from PIL import Image, ImageDraw from PIL import Image
from minio import Minio from minio import Minio
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from fastapi import Response # 导入 FastAPI 的 Response
from config import settings from config import settings
from segment_anything import SamPredictor, sam_model_registry from segment_anything import SamPredictor, sam_model_registry
@@ -21,7 +18,8 @@ minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secre
class SAMRequest(BaseModel): class SAMRequest(BaseModel):
user_id: int bucket: str = Field(...)
object_name: str = Field(...)
image_path: str = Field(..., description="图片路径，必填字段") image_path: str = Field(..., description="图片路径，必填字段")
type: str = Field(..., description="推理类型，必填字段") type: str = Field(..., description="推理类型，必填字段")
points: Optional[List[List[float]]] | None = None points: Optional[List[List[float]]] | None = None