yolo

type Box struct {
	X1 float32 `json:"x1"` // Top-left corner X coordinate
	Y1 float32 `json:"y1"` // Top-left corner Y coordinate
	X2 float32 `json:"x2"` // Bottom-right corner X coordinate
	Y2 float32 `json:"y2"` // Bottom-right corner Y coordinate
}

type DetectOptions struct {
	ConfidenceThreshold float32 `json:"confidence_threshold"` // Confidence threshold for detections
	IoUThreshold        float32 `json:"iou_threshold"`        // IoU threshold for NMS
}

type Detection struct {
	Class      int     `json:"class"`      // Class index
	Label      string  `json:"label"`      // Class label
	Confidence float32 `json:"confidence"` // Confidence score
	Box        Box     `json:"box"`        // Bounding box
}

type DetectionResult struct {
	Detections      []Detection   `json:"detections"`       // Detected objects
	PreprocessTime  time.Duration `json:"preprocess_time"`  // Preprocessing time
	InferenceTime   time.Duration `json:"inference_time"`   // Inference time
	PostprocessTime time.Duration `json:"postprocess_time"` // Postprocessing time
	DetectionTime   time.Duration `json:"detection_time"`   // Total detection time
}

type Detector struct {
	// contains filtered or unexported fields
}

type DetectorOptions struct {
	ModelPath      string           // Path to the ONNX model file
	InterOpThreads int              // Number of inter-op threads
	IntraOpThreads int              // Number of intra-op threads
	CpuMemArena    bool             // Enable CPU memory arena
	MemPattern     bool             // Enable memory pattern optimization
	Provider       *ProviderOptions // Execution provider options
}

type ProviderOptions struct {
	Provider ProviderType

	// CUDA specific options
	CUDADeviceID   int
	CUDAArenaSize  int64
	CUDAStreamSize int64

	// CoreML specific options
	CoreMLFlags uint32

	// TensorRT specific options
	TensorRTDeviceID  int
	TensorRTWorkspace int64
	TensorRTOptLevel  int

	// DirectML specific options
	DirectMLDeviceID int
}

type ProviderType int