From 8c2e1461369b218bb8b5dc042266579ec06a1291 Mon Sep 17 00:00:00 2001 From: jwkim Date: Mon, 10 Nov 2025 10:49:32 +0900 Subject: [PATCH] =?UTF-8?q?edit:=20=EA=B8=B0=EB=B3=B8=EB=AA=A8=EB=8D=B8?= =?UTF-8?q?=ED=83=80=EC=9E=85=20b32=EB=A1=9C=20=EB=B3=80=EA=B2=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main_rest/app/models.py | 4 ++-- main_rest/app/routes/services.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/main_rest/app/models.py b/main_rest/app/models.py index 45ce78d..d3c2ab3 100644 --- a/main_rest/app/models.py +++ b/main_rest/app/models.py @@ -611,7 +611,7 @@ class VectorImageSearchVitReq(BaseModel): ### [Request] vector image search vit """ prompt : str = Field(description='프롬프트', example='검은색 안경') - modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14) + modelType : str = Field(VitModelType.b32, description='pretrained model 타입', example=VitModelType.b32) indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2) searchNum : int = Field(4, description='검색결과 이미지 갯수', example=4) @@ -621,7 +621,7 @@ class VectorImageSearchVitInputImgReq(BaseModel): ### [Request] vector image search vit - input image """ inputImage : str = Field(description='base64 이미지', example='') - modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14) + modelType : str = Field(VitModelType.b32, description='pretrained model 타입', example=VitModelType.b32) indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2) searchNum : int = Field(4, description='검색결과 이미지 갯수', example=4) diff --git a/main_rest/app/routes/services.py b/main_rest/app/routes/services.py index 1218c96..b1208bd 100644 --- a/main_rest/app/routes/services.py +++ b/main_rest/app/routes/services.py @@ -485,7 +485,7 @@ async def vactor_vit_input_img_data(request: Request, request_body_info: M.Vecto > - queryImage는 Input시 입력한 inputImage 이미지 데이터 ### Options - > - modelType -> b32, b16, l14, l14_336 (기본값: l14) + > - modelType -> b32, b16, l14, l14_336 (기본값: b32) > - indexType -> l2, cos (기본값: l2) > - searchNum -> 결과이미지 갯수 (기본값: 4)