sander/go-opencv
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update REAME, split tests.

Browse source
This commit is contained in:
Chih-Wei Chang 2015-02-16 17:07:14 +08:00
parent c4a0fdeeab
commit 15a1472425
3 changed files with 70 additions and 60 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
README.md
View file

@ -49,22 +49,25 @@ Please also note that the basic data structures in OpenCV (e.g., `cv::Mat`, `cv:
```go ```go
package main package main
import "github.com/lazywei/go-opencv/gocv" import . "github.com/lazywei/go-opencv/gocv"
import "github.com/gonum/matrix/mat64"
func main() { func main() {
objPts := gocv.NewGcvPoint3fVector(int64(4))
objPts.Set(0, gocv.NewGcvPoint3f(0, 25, 0))
objPts.Set(1, gocv.NewGcvPoint3f(0, -25, 0))
objPts.Set(2, gocv.NewGcvPoint3f(-47, 25, 0))
objPts.Set(3, gocv.NewGcvPoint3f(-47, -25, 0))
imgPts := gocv.NewGcvPoint2fVector(int64(4)) objPts := mat64.NewDense(4, 3, []float64{
imgPts.Set(0, gocv.NewGcvPoint2f(1136.4140625, 1041.89208984)) 0, 25, 0,
imgPts.Set(1, gocv.NewGcvPoint2f(1845.33190918, 671.39581299)) 0, -25, 0,
imgPts.Set(2, gocv.NewGcvPoint2f(302.73373413, 634.79998779)) -47, 25, 0,
imgPts.Set(3, gocv.NewGcvPoint2f(1051.46154785, 352.76107788)) -47, -25, 0})
gocv.GcvInitCameraMatrix2D(objPts, imgPts) imgPts := mat64.NewDense(4, 2, []float64{
1136.4140625, 1041.89208984,
1845.33190918, 671.39581299,
302.73373413, 634.79998779,
1051.46154785, 352.76107788})
camMat := GcvInitCameraMatrix2D(objPts, imgPts)
fmt.Println(camMat)
} }
``` ```

55
gocv/gocv_calib3d_test.go Normal file
View file

@ -0,0 +1,55 @@
package gocv
import (
"testing"
"github.com/davecgh/go-spew/spew"
"github.com/gonum/matrix/mat64"
)
func TestGcvInitCameraMatrix2D(t *testing.T) {
objPts := mat64.NewDense(4, 3, []float64{
0, 25, 0,
0, -25, 0,
-47, 25, 0,
-47, -25, 0})
imgPts := mat64.NewDense(4, 2, []float64{
1136.4140625, 1041.89208984,
1845.33190918, 671.39581299,
302.73373413, 634.79998779,
1051.46154785, 352.76107788})
camMat := GcvInitCameraMatrix2D(objPts, imgPts)
spew.Dump(camMat)
}
// func TestGcvCalibrateCamera(t *testing.T) {
// objPts := NewGcvPoint3fVector(int64(4))
// objPts.Set(0, NewGcvPoint3f(0, 25, 0))
// objPts.Set(1, NewGcvPoint3f(0, -25, 0))
// objPts.Set(2, NewGcvPoint3f(-47, 25, 0))
// objPts.Set(3, NewGcvPoint3f(-47, -25, 0))
// imgPts := NewGcvPoint2fVector(int64(4))
// imgPts.Set(0, NewGcvPoint2f(1136.4140625, 1041.89208984))
// imgPts.Set(1, NewGcvPoint2f(1845.33190918, 671.39581299))
// imgPts.Set(2, NewGcvPoint2f(302.73373413, 634.79998779))
// imgPts.Set(3, NewGcvPoint2f(1051.46154785, 352.76107788))
// imgSize := NewGcvSize2i(1920, 1080)
// camMat := GcvInitCameraMatrix2D(objPts, imgPts)
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(0, 0)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(0, 1)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(1, 1)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(1, 2)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(2, 2)))
// rvec := NewMat()
// tvec := NewMat()
// GcvCalibrateCamera(objPts, imgPts, imgSize, camMat, rvec, tvec)
// MatToMat64(camMat)
// }

48
gocv/gocv_test.go
View file

@ -4,7 +4,6 @@ import (
"testing" "testing"
"github.com/davecgh/go-spew/spew" "github.com/davecgh/go-spew/spew"
"github.com/gonum/matrix/mat64"
) )
func TestNewGcvPoint3f32(t *testing.T) { func TestNewGcvPoint3f32(t *testing.T) {
@ -27,50 +26,3 @@ func TestMat(t *testing.T) {
mat2 := NewMat(mat) mat2 := NewMat(mat)
spew.Dump(mat2) spew.Dump(mat2)
} }
func TestGcvInitCameraMatrix2D(t *testing.T) {
objPts := mat64.NewDense(4, 3, []float64{
0, 25, 0,
0, -25, 0,
-47, 25, 0,
-47, -25, 0})
imgPts := mat64.NewDense(4, 2, []float64{
1136.4140625, 1041.89208984,
1845.33190918, 671.39581299,
302.73373413, 634.79998779,
1051.46154785, 352.76107788})
camMat := GcvInitCameraMatrix2D(objPts, imgPts)
spew.Dump(camMat)
}
// func TestGcvCalibrateCamera(t *testing.T) {
// objPts := NewGcvPoint3fVector(int64(4))
// objPts.Set(0, NewGcvPoint3f(0, 25, 0))
// objPts.Set(1, NewGcvPoint3f(0, -25, 0))
// objPts.Set(2, NewGcvPoint3f(-47, 25, 0))
// objPts.Set(3, NewGcvPoint3f(-47, -25, 0))
// imgPts := NewGcvPoint2fVector(int64(4))
// imgPts.Set(0, NewGcvPoint2f(1136.4140625, 1041.89208984))
// imgPts.Set(1, NewGcvPoint2f(1845.33190918, 671.39581299))
// imgPts.Set(2, NewGcvPoint2f(302.73373413, 634.79998779))
// imgPts.Set(3, NewGcvPoint2f(1051.46154785, 352.76107788))
// imgSize := NewGcvSize2i(1920, 1080)
// camMat := GcvInitCameraMatrix2D(objPts, imgPts)
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(0, 0)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(0, 1)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(1, 1)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(1, 2)))
// spew.Dump(camMat.GcvAtd(NewGcvSize2i(2, 2)))
// rvec := NewMat()
// tvec := NewMat()
// GcvCalibrateCamera(objPts, imgPts, imgSize, camMat, rvec, tvec)
// MatToMat64(camMat)
// }