mirror of
https://github.com/photoprism/photoprism.git
synced 2025-09-26 21:01:58 +08:00
62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/photoprism/photoprism/internal/ai/vision"
|
|
"github.com/photoprism/photoprism/internal/config"
|
|
"github.com/photoprism/photoprism/internal/entity"
|
|
"github.com/photoprism/photoprism/internal/entity/search"
|
|
"github.com/photoprism/photoprism/internal/workers"
|
|
)
|
|
|
|
// VisionRunCommand configures the command name, flags, and action.
|
|
var VisionRunCommand = &cli.Command{
|
|
Name: "run",
|
|
Usage: "Runs one or more computer vision models on a set of pictures that match the specified search filters",
|
|
ArgsUsage: "[filter]...",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "models",
|
|
Aliases: []string{"m"},
|
|
Usage: "computer vision `MODELS` to run, e.g. caption, labels, or nsfw",
|
|
Value: "caption",
|
|
},
|
|
&cli.IntFlag{
|
|
Name: "count",
|
|
Aliases: []string{"n"},
|
|
Usage: "maximum `NUMBER` of pictures to be processed",
|
|
Value: search.MaxResults,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "source",
|
|
Aliases: []string{"s"},
|
|
Value: entity.SrcImage,
|
|
Usage: "custom data source `ROLE` e.g. default, image, meta, vision, or admin",
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "force",
|
|
Aliases: []string{"f"},
|
|
Usage: "replaces existing data if the model supports it and the source priority is equal or higher",
|
|
},
|
|
},
|
|
Action: visionRunAction,
|
|
}
|
|
|
|
// visionListAction displays existing user accounts.
|
|
func visionRunAction(ctx *cli.Context) error {
|
|
return CallWithDependencies(ctx, func(conf *config.Config) error {
|
|
worker := workers.NewVision(conf)
|
|
filter := strings.TrimSpace(strings.Join(ctx.Args().Slice(), " "))
|
|
return worker.Start(
|
|
filter,
|
|
ctx.Int("count"),
|
|
vision.ParseTypes(ctx.String("models")),
|
|
ctx.String("source"),
|
|
ctx.Bool("force"),
|
|
)
|
|
})
|
|
}
|