44 lines
960 B
Go
44 lines
960 B
Go
package main
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/airbusgeo/godal"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"gocv.io/x/gocv"
|
|
"starwiz.cn/sjy01/image-proc/pkg/producer"
|
|
"starwiz.cn/sjy01/image-proc/pkg/utils"
|
|
)
|
|
|
|
var (
|
|
input0 string
|
|
)
|
|
|
|
var testCmd = &cobra.Command{
|
|
Use: "test",
|
|
Short: "test command",
|
|
Long: `test command`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
godal.RegisterAll()
|
|
|
|
input0_mat, err := utils.ReadTifftoMat(input0)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return
|
|
}
|
|
|
|
mv := gocv.Split(*input0_mat)
|
|
for i := 0; i < len(mv); i++ {
|
|
out := producer.CV_Sobel(mv[i])
|
|
utils.SavePanToGDALGTiff(out, 0, 0, "data/temp/out_"+strconv.Itoa(i)+".tif", 1.3)
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(testCmd)
|
|
|
|
testCmd.Flags().StringVarP(&input0, "input0", "i", "/Users/lan/workspace/sjy01/image-proc/data/002438/L1A/ROI_7000_10500/MSS/SJY01_MSS_20241014_121246_002438_018_ROI_7000_10500_L1A.tiff", "input0")
|
|
}
|