Files
slogan/styleagent/agent/avatar_matcher.go
T

43 lines
1.1 KiB
Go

package agent
// FaceFeature 从照片+用户填写提取的化身特征(v1:肤色/身高/体重来自身形参数,照片贴图后续增强)
type FaceFeature struct {
SkinTone int // 1-5
HeightCm int
WeightKg int
}
// 预烘焙模板库索引(构建期产物,运行时只读常量)
const (
FaceTemplateCount = 20
BodyTemplateCount = 6
SkinToneLevels = 5
DefaultFaceTemplate = 5
DefaultBodyTemplate = 3
)
// MatchTemplates 特征 → 模板索引
// 身体模板:身高 145-190cm 映射 6 档;肤色 1-5 直接映射皮肤贴图档
func MatchTemplates(f *FaceFeature) (faceId, bodyId, skinIdx int) {
if f == nil {
return DefaultFaceTemplate, DefaultBodyTemplate, 3
}
skinIdx = f.SkinTone
if skinIdx < 1 {
skinIdx = 1
}
if skinIdx > SkinToneLevels {
skinIdx = SkinToneLevels
}
bodyId = (f.HeightCm-145)/8 + 1
if bodyId < 1 {
bodyId = 1
}
if bodyId > BodyTemplateCount {
bodyId = BodyTemplateCount
}
// v1 脸型固定默认模板(AI 人脸特征提取后替换,见 spec v2)
faceId = DefaultFaceTemplate
return
}