edit : 이미지생성 api 생성갯수 삭제, 이미지생성 api 데이터를 return해주는 api 추가, 입력이미지 입력받아 검색하는 api 추가

This commit is contained in:
2025-08-07 16:54:46 +09:00
parent 5c20b9ccb5
commit 4ef360448a
5 changed files with 151 additions and 22 deletions

View File

@@ -611,6 +611,7 @@ class VactorSearchVitReq(BaseModel):
model_type : VitModelType = Field(VitModelType.l14, description='pretrained 모델 정보', example=VitModelType.l14)
search_num : int = Field(4, description='검색결과 이미지 갯수', example=4)
class VactorSearchVitRes(ResponseBase):
img_list : dict = Field({}, description='이미지 결과 리스트', example={})

View File

@@ -55,6 +55,9 @@ async def vactor_search(request: Request, request_body_info: M.VactorSearchReq):
@router.post("/faiss/vector/search/vit/report", summary="vit search report", response_model=M.ResponseBase)
async def vactor_report_vit(request: Request, request_body_info: M.VactorSearchVitReportReq):
"""
이미지 경로를 입력받아 repport image 저장
"""
response = M.ResponseBase()
try:
if not os.path.exists(request_body_info.query_image_path):
@@ -74,6 +77,9 @@ async def vactor_report_vit(request: Request, request_body_info: M.VactorSearchV
@router.post("/faiss/vector/search/vit", summary="vit search", response_model=M.VactorSearchVitRes)
async def vactor_vit(request: Request, request_body_info: M.VactorSearchVitReq):
"""
이미지 경로 를 입력받아 vit 방식으로 검색
"""
response = M.VactorSearchVitRes()
try:
if not os.path.exists(request_body_info.query_image_path):