32 lines
751 B
Go
32 lines
751 B
Go
package main
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"starwiz.cn/sjy01/image-proc/pkg/rrc"
|
|
)
|
|
|
|
var (
|
|
panDS string
|
|
mssDS string
|
|
lutOut string
|
|
)
|
|
|
|
var rrcCmd = &cobra.Command{
|
|
Use: "rrc",
|
|
Short: "Run RRC algorithm on an image",
|
|
Long: `Run RRC algorithm on an image`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
rrc := rrc.NewRRC(lutOut)
|
|
rrc.StatisticalPAN(panDS)
|
|
rrc.StatisticalMSS(mssDS)
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rrcCmd.Flags().StringVarP(&panDS, "pan-dataset", "p", "data/RAW/pan.txt", "path to pan dataset")
|
|
rrcCmd.Flags().StringVarP(&mssDS, "mss-dataset", "m", "data/RAW/mss-none.txt", "path to mss dataset")
|
|
rrcCmd.Flags().StringVarP(&lutOut, "lut-out-dir", "o", "data/rrc", "path to output lut file")
|
|
|
|
rootCmd.AddCommand(rrcCmd)
|
|
}
|