修改创建实例方式

This commit is contained in:
zqc
2025-12-21 12:00:04 +08:00
parent 76895da660
commit 7d79862981
2 changed files with 4 additions and 10 deletions

View File

@@ -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:
# 转换为二进制数据

View File

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