66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package photoprism
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/photoprism/photoprism/internal/entity"
|
|
"github.com/photoprism/photoprism/internal/face"
|
|
"github.com/photoprism/photoprism/internal/query"
|
|
)
|
|
|
|
// FacesOptimizeResult represents the outcome of Faces.Optimize().
|
|
type FacesOptimizeResult struct {
|
|
Merged int
|
|
}
|
|
|
|
// Optimize optimizes the face lookup table.
|
|
func (w *Faces) Optimize() (result FacesOptimizeResult, err error) {
|
|
if w.Disabled() {
|
|
return result, fmt.Errorf("face recognition is disabled")
|
|
}
|
|
|
|
// Iterative merging of manually added face clusters.
|
|
for i := 0; i <= 10; i++ {
|
|
var n int
|
|
var c = result.Merged
|
|
var merge entity.Faces
|
|
var faces entity.Faces
|
|
|
|
// Fetch manually added faces from the database.
|
|
if faces, err = query.ManuallyAddedFaces(false, face.RegularFace); err != nil {
|
|
return result, err
|
|
} else if n = len(faces) - 1; n < 1 {
|
|
// Need at least 2 faces to optimize.
|
|
break
|
|
}
|
|
|
|
// Find and merge matching faces.
|
|
for j := 0; j <= n; j++ {
|
|
if len(merge) == 0 {
|
|
merge = entity.Faces{faces[j]}
|
|
} else if faces[j].SubjUID != merge[len(merge)-1].SubjUID || j == n {
|
|
if len(merge) < 2 {
|
|
// Nothing to merge.
|
|
} else if _, err := query.MergeFaces(merge); err != nil {
|
|
log.Errorf("%s (merge)", err)
|
|
} else {
|
|
result.Merged += len(merge)
|
|
}
|
|
|
|
merge = nil
|
|
} else if ok, dist := merge[0].Match(face.Embeddings{faces[j].Embedding()}); ok {
|
|
log.Debugf("faces: can merge %s with %s, subject %s, dist %f", merge[0].ID, faces[j].ID, entity.SubjNames.Log(merge[0].SubjUID), dist)
|
|
merge = append(merge, faces[j])
|
|
} else if len(merge) == 1 {
|
|
merge = nil
|
|
}
|
|
}
|
|
|
|
// Done?
|
|
if result.Merged <= c {
|
|
break
|
|
}
|
|
}
|
|
|
|
return result, nil
|
|
}
|