-
-
Notifications
You must be signed in to change notification settings - Fork 82
Open
Description
Inference result tensor data is 0 value
`package main
import (
"fmt"
"image"
"os"
"github.com/owulveryck/onnx-go"
"github.com/owulveryck/onnx-go/backend/x/gorgonnx"
"gocv.io/x/gocv"
"gorgonia.org/tensor"
)
func main() {
run()
}
func run() {
mfile, _ := os.ReadFile("/home/jetson/onnx-go/yolov10s.onnx")
backend := gorgonnx.NewGraph()
model := onnx.NewModel(backend)
model.UnmarshalBinary(mfile)
mat := gocv.IMRead("/home/jetson/onnx-go/bus.jpg", gocv.IMReadColor)
defer mat.Close()
blob := gocv.BlobFromImage(mat, 1.0/255, image.Point{640, 640}, gocv.NewScalar(0, 0, 0, 0), true, false)
inputdata, _ := blob.DataPtrFloat32()
inputtensor := tensor.New(tensor.Of(tensor.Float32), tensor.WithShape(1, 3, 640, 640), tensor.WithBacking(inputdata))
model.SetInput(0, inputtensor)
backend.Run()
output, _ := model.GetOutputTensors()
a := output[0].Data()
fmt.Println(a.([]float32))
}`
`panic: onnx: operator ConstantOfShape not implemented ()
goroutine 1 [running]:
main.run()
/home/jetson/onnx-go/test1/main.go:54 +0x5b8
main.main()
/home/jetson/onnx-go/test1/main.go:15 +0x1c
exit status 2`
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels