.

cnn_router.py

@@ -76,10 +76,14 @@ async def extract_features(files: List[UploadFile] = File(...)):
             
             # Inference
             with torch.no_grad():
-                embedding = vision_model(input_tensor)
+                raw_embedding = vision_model(input_tensor)
+                
+                # [추가 작업] L2 정규화 (Normalize)
+                # p=2는 L2 norm을 의미, dim=1은 차원 방향
+                normalized_embedding = F.normalize(raw_embedding, p=2, dim=1)
             
             # Convert to list
-            embedding_list = embedding.squeeze().tolist()
+            embedding_list = normalized_embedding.squeeze().tolist()
             
             result_data.append({
                 "key": file.filename,