edit : 이미지생성 api 생성갯수 삭제, 이미지생성 api 데이터를 return해주는 api 추가, 입력이미지 입력받아 검색하는 api 추가
This commit is contained in:
@@ -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={})
|
||||
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user