FERMAT_TEST_REST_SERVER/GLASSES_AI_SERVER
4
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

edit: 기본모델타입 b32로 변경

Browse Source
This commit is contained in:
김재웅
2025-11-10 10:49:32 +09:00
parent 55d92878d5
commit 8c2e146136
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main_rest/app/models.py
View File

@@ -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)

2
main_rest/app/routes/services.py
View File

@@ -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)