faces_optimize.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, false); 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, false); 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
}