42 lines
881 B
Go
42 lines
881 B
Go
|
package bilds
|
||
|
|
|
||
|
|
import (
|
||
|
|
"image"
|
||
|
|
|
||
|
|
"github.com/anthonynsimon/bild/segment"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
func threshold() *cobra.Command {
|
||
|
|
var level uint8
|
||
|
|
|
||
|
|
var cmd = &cobra.Command{
|
||
|
|
Use: "threshold",
|
||
|
|
Short: "segment an image by a threshold",
|
||
|
|
Args: cobra.ExactArgs(2),
|
||
|
|
Example: "threshold --level 200 input.jpg output.jpg",
|
||
|
|
Run: func(cmd *cobra.Command, args []string) {
|
||
|
|
fin := args[0]
|
||
|
|
fout := args[1]
|
||
|
|
|
||
|
|
apply(fin, fout, func(img image.Image) (image.Image, error) {
|
||
|
|
return segment.Threshold(img, level), nil
|
||
|
|
})
|
||
|
|
}}
|
||
|
|
|
||
|
|
cmd.Flags().Uint8VarP(&level, "level", "l", 128, "the level at which the segmenting threshold will be crossed")
|
||
|
|
|
||
|
|
return cmd
|
||
|
|
}
|
||
|
|
|
||
|
|
func createSegment() *cobra.Command {
|
||
|
|
var blurCmd = &cobra.Command{
|
||
|
|
Use: "segment",
|
||
|
|
Short: "segment an image using the specified method",
|
||
|
|
}
|
||
|
|
|
||
|
|
blurCmd.AddCommand(threshold())
|
||
|
|
|
||
|
|
return blurCmd
|
||
|
|
}
|