mirror of
https://github.com/photoprism/photoprism.git
synced 2025-12-12 00:34:13 +01:00
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package vision
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/photoprism/photoprism/pkg/media/http/scheme"
|
|
)
|
|
|
|
// Default computer vision model configuration.
|
|
var (
|
|
NasnetModel = &Model{
|
|
Type: ModelTypeLabels,
|
|
Name: "NASNet",
|
|
Version: "Mobile",
|
|
Resolution: 224,
|
|
Tags: []string{"photoprism"},
|
|
}
|
|
NsfwModel = &Model{
|
|
Type: ModelTypeNsfw,
|
|
Name: "Nsfw",
|
|
Version: "",
|
|
Resolution: 224,
|
|
Tags: []string{"serve"},
|
|
}
|
|
FacenetModel = &Model{
|
|
Type: ModelTypeFace,
|
|
Name: "FaceNet",
|
|
Version: "",
|
|
Resolution: 160,
|
|
Tags: []string{"serve"},
|
|
}
|
|
CaptionModel = &Model{
|
|
Type: ModelTypeCaption,
|
|
Resolution: 224,
|
|
Name: CaptionModelDefault,
|
|
Version: ModelVersionDefault,
|
|
Prompt: CaptionPromptDefault,
|
|
Service: Service{
|
|
Uri: "http://photoprism-vision:5000/api/v1/vision/caption",
|
|
Method: http.MethodPost,
|
|
FileScheme: scheme.Data,
|
|
RequestFormat: ApiFormatVision,
|
|
ResponseFormat: ApiFormatVision,
|
|
},
|
|
}
|
|
DefaultModels = Models{NasnetModel, NsfwModel, FacenetModel, CaptionModel}
|
|
DefaultThresholds = Thresholds{Confidence: 10}
|
|
)
|