diff --git a/src/api/routes/algorithm_router.py b/src/api/routes/algorithm_router.py index 651270b..b9eb220 100644 --- a/src/api/routes/algorithm_router.py +++ b/src/api/routes/algorithm_router.py @@ -16,6 +16,7 @@ from src.database.connection import db_manager from src.models.face_feature import SurFaceFeature, FeatureStatus from src.repositories.face_feature_repository import FaceFeatureRepository from src.face_recognition_algorithm import FaceRecognitionAlgorithm +from src.base_face_biz import BaseFaceBiz # 创建路由器 router = APIRouter(prefix="/algorithm", tags=["algorithm"]) @@ -95,7 +96,9 @@ def process_feature_calculation(feature_id: int) -> bool: # 提取人脸特征 try: - feature_vector = face_algorithm.create_business().extract_face_feature(str(image_path)) + # 直接创建BaseFaceBiz实例 + face_biz = BaseFaceBiz(face_algorithm.get_app()) + feature_vector = face_biz.extract_face_feature(str(image_path)) if feature_vector is not None: # 转换为二进制数据 diff --git a/src/face_recognition_algorithm.py b/src/face_recognition_algorithm.py index e335dd8..23927c4 100644 --- a/src/face_recognition_algorithm.py +++ b/src/face_recognition_algorithm.py @@ -75,15 +75,6 @@ class FaceRecognitionAlgorithm: self.det_size = size print(f"✅ 检测尺寸设置为: {size}") - def create_business(self) -> BaseFaceBiz: - """ - 创建业务实例 - - 返回: - BaseFaceBiz实例,用于处理具体的业务逻辑 - """ - return BaseFaceBiz(self.app) - def get_app(self) -> FaceAnalysis: """获取底层的FaceAnalysis实例""" return self.app \ No newline at end of file