mirror of
https://github.com/hybridgroup/gocv
synced 2025-08-25 08:41:04 +08:00
121 lines
2.3 KiB
Go
121 lines
2.3 KiB
Go
package cuda
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gocv.io/x/gocv"
|
|
)
|
|
|
|
func TestNewGpuMat(t *testing.T) {
|
|
mat := NewGpuMat()
|
|
defer mat.Close()
|
|
|
|
if !mat.Empty() {
|
|
t.Error("New GpuMat should be empty")
|
|
}
|
|
}
|
|
|
|
func TestGpuMatClosed(t *testing.T) {
|
|
mat := NewGpuMat()
|
|
mat.Close()
|
|
|
|
if !mat.Closed() {
|
|
t.Error("Closed GpuMat should be closed")
|
|
}
|
|
}
|
|
|
|
func TestNewGpuMatFromMat(t *testing.T) {
|
|
mat := gocv.NewMat()
|
|
defer mat.Close()
|
|
|
|
gpumat := NewGpuMatFromMat(mat)
|
|
defer gpumat.Close()
|
|
|
|
if !gpumat.Empty() {
|
|
t.Error("New GpuMat should be empty")
|
|
}
|
|
}
|
|
|
|
func TestNewGpuMatFromMatWithSize(t *testing.T) {
|
|
mat := gocv.NewMatWithSize(100, 200, gocv.MatTypeCV32FC4)
|
|
defer mat.Close()
|
|
|
|
gpumat := NewGpuMatFromMat(mat)
|
|
defer gpumat.Close()
|
|
|
|
if gpumat.Empty() {
|
|
t.Error("New GpuMat should be not empty")
|
|
}
|
|
|
|
if gpumat.Rows() != 100 {
|
|
t.Error("incorrect number of rows for GpuMat")
|
|
}
|
|
|
|
if gpumat.Cols() != 200 {
|
|
t.Error("incorrect number of cols for GpuMat")
|
|
}
|
|
|
|
if gpumat.Type() != gocv.MatTypeCV32FC4 {
|
|
t.Error("incorrect type for GpuMat")
|
|
}
|
|
}
|
|
|
|
func TestNewGpuMatWithSize(t *testing.T) {
|
|
gpumat := NewGpuMatWithSize(100, 200, gocv.MatTypeCV32FC4)
|
|
defer gpumat.Close()
|
|
|
|
if gpumat.Empty() {
|
|
t.Error("New GpuMat should be not empty")
|
|
}
|
|
|
|
if gpumat.Rows() != 100 {
|
|
t.Error("incorrect number of rows for GpuMat")
|
|
}
|
|
|
|
if gpumat.Cols() != 200 {
|
|
t.Error("incorrect number of cols for GpuMat")
|
|
}
|
|
|
|
if gpumat.Type() != gocv.MatTypeCV32FC4 {
|
|
t.Error("incorrect type for GpuMat")
|
|
}
|
|
}
|
|
|
|
func TestGetCudaEnabledDeviceCount(t *testing.T) {
|
|
if GetCudaEnabledDeviceCount() < 1 {
|
|
t.Fatal("expected atleast one cuda enabled device")
|
|
}
|
|
}
|
|
|
|
func TestConvertFp16(t *testing.T) {
|
|
gpumat := NewGpuMatWithSize(100, 200, gocv.MatTypeCV32FC4)
|
|
defer gpumat.Close()
|
|
|
|
fp16mat := NewGpuMatWithSize(100, 200, gocv.MatTypeCV32FC4)
|
|
defer fp16mat.Close()
|
|
|
|
gpumat.ConvertFp16(&fp16mat)
|
|
|
|
if fp16mat.Empty() {
|
|
t.Error("New fp16mat should be not empty")
|
|
}
|
|
|
|
if fp16mat.Rows() != 100 {
|
|
t.Error("incorrect number of rows for fp16mat")
|
|
}
|
|
|
|
if fp16mat.Cols() != 200 {
|
|
t.Error("incorrect number of cols for fp16mat")
|
|
}
|
|
|
|
if fp16mat.Type() != gocv.MatTypeCV16SC4 {
|
|
t.Error("incorrect type for fp16mat", fp16mat.Type())
|
|
}
|
|
}
|
|
|
|
func TestCudaDeviceSupports(t *testing.T) {
|
|
if !DeviceSupports(FeatureSetCompute10) {
|
|
t.Fatal("expected FeatureSetCompute10 on cuda enabled device")
|
|
}
|
|
}
|