优化

38adf698 · zhengyaoqiu · be610aa6 · 38adf698
Commit 38adf698 authored May 30, 2025 by zhengyaoqiu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

feature_extractor.py app/services/feature_extractor.py +2 -1

No files found.
--- a/app/services/feature_extractor.py
+++ b/app/services/feature_extractor.py
@@ -115,7 +115,8 @@ class FeatureExtractor:

                result = image_features.cpu().numpy().astype(np.float32).flatten()
                # ViT-L/14@336px 需要转换列表结构
-                return result.tolist()
+                # return result.tolist()
+                return result

        except Exception as e:
            self.__logger.error(f"Error extracting features from image: {e}")