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)