43 lines
828 B
Go
43 lines
828 B
Go
package producer
|
|
|
|
import (
|
|
"image"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"gocv.io/x/gocv"
|
|
)
|
|
|
|
type PhaseShiftM struct {
|
|
dx float32
|
|
dy float32
|
|
response float64
|
|
Block Block
|
|
}
|
|
|
|
type Block struct {
|
|
width int
|
|
height int
|
|
coord image.Point // top-left corner of the block in the original image
|
|
}
|
|
|
|
func (r *Registrator) calculateBlockPhaseShift(panBlock, mssBlock gocv.Mat) (gocv.Point2f, float64) {
|
|
pan := gocv.NewMat()
|
|
mss := gocv.NewMat()
|
|
|
|
panBlock.ConvertTo(&pan, gocv.MatTypeCV32FC1)
|
|
defer pan.Close()
|
|
mssBlock.ConvertTo(&mss, gocv.MatTypeCV32FC1)
|
|
defer mss.Close()
|
|
|
|
hann := gocv.NewMat()
|
|
defer hann.Close()
|
|
|
|
shift, response := gocv.PhaseCorrelate(pan, mss, hann)
|
|
|
|
dx := shift.X
|
|
dy := shift.Y
|
|
log.Debugf("Block shift: dx = %f, dy = %f. response = %f", dx, dy, response)
|
|
|
|
return shift, response
|
|
}
|