mirror of
https://github.com/AndreyGermanov/yolov8_onnx_go.git
synced 2025-09-27 04:26:22 +08:00
17 lines
286 B
Go
17 lines
286 B
Go
package main
|
|
|
|
import ort "github.com/yalue/onnxruntime_go"
|
|
|
|
var (
|
|
UseCoreML = false
|
|
Blank []float32
|
|
ModelPath = "./yolov8m.onnx"
|
|
Yolo8Model ModelSession
|
|
)
|
|
|
|
type ModelSession struct {
|
|
Session *ort.AdvancedSession
|
|
Input *ort.Tensor[float32]
|
|
Output *ort.Tensor[float32]
|
|
}
|