✨ add image clarity detection

app/aisvc/rpc/internal/logic/aiservice/image_clarity_logic.go

@@ -0,0 +1,36 @@
+package aiservicelogic
+
+import (
+	"context"
+
+	"schisandra-album-cloud-microservices/app/aisvc/rpc/internal/svc"
+	"schisandra-album-cloud-microservices/app/aisvc/rpc/pb"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type ImageClarityLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewImageClarityLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ImageClarityLogic {
+	return &ImageClarityLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// ImageClarity 图像清晰度检测
+func (l *ImageClarityLogic) ImageClarity(in *pb.ImageClarityRequest) (*pb.ImageClarityResponse, error) {
+	blurred, confidence, err := l.svcCtx.Clarity.Detect(in.Image)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.ImageClarityResponse{
+		IsBlurred:  blurred,
+		Confidence: float32(confidence),
+	}, nil
+}

app/aisvc/rpc/internal/server/aiservice/ai_service_server.go

@@ -58,3 +58,9 @@ func (s *AiServiceServer) ModifyFaceType(ctx context.Context, in *pb.ModifyFaceT
 	l := aiservicelogic.NewModifyFaceTypeLogic(ctx, s.svcCtx)
 	return l.ModifyFaceType(in)
 }
+
+// FuzzySearch
+func (s *AiServiceServer) ImageClarity(ctx context.Context, in *pb.ImageClarityRequest) (*pb.ImageClarityResponse, error) {
+	l := aiservicelogic.NewImageClarityLogic(ctx, s.svcCtx)
+	return l.ImageClarity(in)
+}

app/aisvc/rpc/internal/svc/service_context.go

@@ -9,6 +9,7 @@ import (
 	"schisandra-album-cloud-microservices/app/aisvc/model/mysql/query"
 	"schisandra-album-cloud-microservices/app/aisvc/rpc/internal/config"
 	"schisandra-album-cloud-microservices/common/caffe_classifier"
+	"schisandra-album-cloud-microservices/common/clarity"
 	"schisandra-album-cloud-microservices/common/face_recognizer"
 	"schisandra-album-cloud-microservices/common/miniox"
 	"schisandra-album-cloud-microservices/common/redisx"
@@ -25,6 +26,7 @@ type ServiceContext struct {
 	CaffeNet       *gocv.Net
 	CaffeDesc      []string
 	MinioClient    *minio.Client
+	Clarity        *clarity.Detector
 }
 
 func NewServiceContext(c config.Config) *ServiceContext {
@@ -42,5 +44,6 @@ func NewServiceContext(c config.Config) *ServiceContext {
 		CaffeNet:       caffeClassifier,
 		CaffeDesc:      caffeDesc,
 		MinioClient:    miniox.NewMinio(c.Minio.Endpoint, c.Minio.AccessKeyID, c.Minio.SecretAccessKey, c.Minio.UseSSL),
+		Clarity:        clarity.NewDetector(clarity.WithConcurrency(8), clarity.WithBaseThreshold(90), clarity.WithEdgeBoost(1.2), clarity.WithSampleScale(1)),
 	}
 }