lankun/sjy01-image-proc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

保留可配置入口,未启用

Browse Source
This commit is contained in:
nuknal
2024-05-30 11:46:50 +08:00
parent 07ee4d88d4
commit e4d6b35702
11 changed files with 159 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/producer/gdal_pansharpen.go
View File

@@ -83,9 +83,8 @@ func PansharpenIHS(panImage, mssImage gocv.Mat) gocv.Mat {
return fused16U
}
func GDALPansharpen(panTiff, mssTiff string) (string, error) {
fusedTIff := strings.Replace(mssTiff, "MSS", "FUS", -1)
os.MkdirAll(filepath.Dir(fusedTIff), 0755)
func GDALPansharpen(panTiff, mssTiff, fusedTiff string) error {
os.MkdirAll(filepath.Dir(fusedTiff), 0755)
// pansharpenCmd := exec.Command("gdal_pansharpen.py", "-w 0.6 -w 0.1 -w 0.3 -w 0 -b 3 -b 2 -b 1", panTiff, mssTiff, fusedTIff, "-r cubic", "-of GTiff")
pansharpenCmd := exec.Command("gdal_pansharpen.py",
"-w", "0.4", "-w", "0.2", "-w", "0.4", "-w", "0",
@@ -93,11 +92,11 @@ func GDALPansharpen(panTiff, mssTiff string) (string, error) {
"-r", "cubic", "-of", "GTiff",
panTiff,
mssTiff,
fusedTIff)
fusedTiff)
log.Println("Run Command: ", pansharpenCmd.String())
pansharpenCmd.Stdout = os.Stdout
pansharpenCmd.Stderr = os.Stderr
return fusedTIff, pansharpenCmd.Run()
return pansharpenCmd.Run()
}
func GDALTranslate(srcTiff, dstJPG string) error {

2
pkg/producer/image_registration.go
View File

@@ -325,7 +325,7 @@ func (r *Registrator) calcDeltaCoeffs() error {
return nil
}
func (r *Registrator) DoCoRegestration() error {
func (r *Registrator) DoCoRegistration() error {
for band := 0; band < MssBands; band++ {
if len(r.deltaXCoeffs[band]) < 2 || len(r.deltaYCoeffs[band]) < 3 {
log.Error("delta coefficients not calculated, skip co-registration")

6
pkg/producer/jpg.go
View File

@@ -131,12 +131,6 @@ func calculateAverageBrightness(img gocv.Mat) float64 {
defer gray.Close()
gocv.CvtColor(img, &gray, gocv.ColorBGRToGray)
return gray.Mean().Val1
// mean := gocv.NewMat()
// dstStdDev := gocv.NewMat()
// defer mean.Close()
// defer dstStdDev.Close()
// gocv.MeanStdDev(gray, &mean, &dstStdDev)
// return float64(mean.GetFloatAt(0, 0))
}
// determineGammaValue determines an appropriate gamma value based on the average brightness

2
pkg/producer/params.go
View File

@@ -5,7 +5,7 @@ type Params struct {
MssRawFile string
AuxRawFile string
SaveRegisteredMssRaw bool
PansharpenIHS bool
DoPansharpen bool
OutputDir string
PanTiffFile string
MssTiffFile string

9
pkg/producer/scenes.go
View File

@@ -131,13 +131,8 @@ func (r *Registrator) SaveScenesToTiff(panScenes []*Scene, mssScenes []*Scene) e
func (r *Registrator) DoScenePansharpen(panScenes []*Scene, mssScenes []*Scene) error {
for i := 0; i < len(panScenes); i++ {
// rgbirImg, _ := r.MergeMSSToBGRNIR(mssScenes[i].Mat)
// ihsImage := PansharpenIHS(panScenes[i].Mat[0], rgbirImg)
// path := strings.TrimSuffix(r.Params.MssTiffFile, ".tiff")
// filename := fmt.Sprintf("%s_%03d_FUS.tiff", path, i+1)
// SaveBGRToGDALGTiff(ihsImage, 3, 1.25, filename)
fusedTiff, err := GDALPansharpen(panScenes[i].Tiff, mssScenes[i].Tiff)
fusedTiff := strings.Replace(mssScenes[i].Tiff, "MSS", "FUS", -1)
err := GDALPansharpen(panScenes[i].Tiff, mssScenes[i].Tiff, fusedTiff)
if err != nil {
return err
}