darknet

type BBOX struct {
	Rect          image.Rectangle // (x,y) - top-left corner, (w, h) - width & height of bounded box
	Confidence    float32         // confidence - probability that the object was found correctly
	ObjId         uint            // class of object - from range [0, classes-1]
	TrackId       uint            // tracking id for video (0 - untracked, 1 - inf - tracked object)
	FramesCounter uint            // counter of frames on which the object was detected
	X3d, Y3d, Z3d float32         // center of object (in Meters) if ZED 3D Camera is used
}

type BoundingBox struct {
	StartPoint image.Point
	EndPoint   image.Point
}

type DarknetImage struct {
	Width  int
	Height int
	// contains filtered or unexported fields
}

type Detection struct {
	BoundingBox

	ClassIDs      []int
	ClassNames    []string
	Probabilities []float32
}

type DetectionResult struct {
	Detections           []*Detection
	NetworkOnlyTimeTaken time.Duration
	OverallTimeTaken     time.Duration
}

type YOLONetwork struct {
	GPUDeviceIndex           int
	NetworkConfigurationFile string
	WeightsFile              string
	Threshold                float32

	ClassNames []string
	Classes    int
	// contains filtered or unexported fields
}