apps/go-rknnlite
1
0
Fork 0
mirror of https://github.com/swdee/go-rknnlite.git synced 2026-04-22 22:57:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix issue on inferfence logic used to calculate tensor size to support 4D (batch) tensors

Browse Source
This commit is contained in:
swdee
2025-06-30 13:27:08 +12:00
parent 51d5c80ce0
commit 207a0a5a41
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
inference.go
+2 -2
View File
@@ -60,7 +60,7 @@ func (r *Runtime) Inference(mats []gocv.Mat) (*Outputs, error) {
Index: uint32(idx),
Type: TensorFloat32,
// multiply by 4 for size of float32
Size: uint32(mat.Cols() * mat.Rows() * mat.Channels() * 4),
Size: uint32(len(data) * 4), // bytes = elements * 4
Fmt: TensorNHWC,
Buf: unsafe.Pointer(&data[0]),
PassThrough: false,
@@ -77,7 +77,7 @@ func (r *Runtime) Inference(mats []gocv.Mat) (*Outputs, error) {
inputs[idx] = Input{
Index: uint32(idx),
Type: TensorUint8,
Size: uint32(mat.Cols() * mat.Rows() * mat.Channels()),
Size: uint32(len(data)), // bytes = elements
Fmt: TensorNHWC,
Buf: unsafe.Pointer(&data[0]),
PassThrough: false,