Why Go
- Case Studies
  
  Common problems companies solve with Go
- Use Cases
  
  Stories about how and why companies use Go
- Security Policy
  
  How Go can help keep you secure by default
Learn
Docs
- Effective Go
  
  Tips for writing clear, performant, and idiomatic Go code
- Go User Manual
  
  A complete introduction to building software with Go
- Standard library
  
  Reference documentation for Go's standard library
- Release Notes
  
  Learn what's new in each Go release
Packages
Community
- Recorded Talks
  
  Videos from prior events
- Meetups
  
  Meet other local Go developers
- Conferences
  
  Learn and network with Go developers from around the world
- Go blog
  
  The Go project's official blog.
- Go project
  
  Get help and stay informed from Go
- Get connected

yolotriton

package module

Go to main page

Versions in this module

v0

v0.4.0

Sep 19, 2023

Changes in this version

+ var YoloClasses = []string

type YoloTritonConfig

+ Classes []string

v0.3.0

Sep 15, 2023

Changes in this version

type Model

+ func NewYoloNASInt8(modelName string, modelVersion string) Model

+ GetClass func(index int) string

type ModelType

+ const ModelTypeYoloNASInt8

type YoloNAS

+ func (y *YoloNAS) GetClass(index int) string

+ type YoloNASInt8 struct

+ func (y *YoloNASInt8) GetClass(index int) string

+ func (y *YoloNASInt8) GetConfig() YoloTritonConfig

+ func (y *YoloNASInt8) PostProcess(rawOutputContents [][]byte) ([]Box, error)

+ func (y *YoloNASInt8) PreProcess(img image.Image, targetWidth uint, targetHeight uint) (*triton.InferTensorContents, error)

type YoloTritonConfig

+ NumClasses int

type YoloV8

+ func (y *YoloV8) GetClass(index int) string

v0.2.0

Sep 6, 2023

Changes in this version

+ type Model interface

+ GetConfig func() YoloTritonConfig

+ PostProcess func(rawOutputContents [][]byte) ([]Box, error)

+ PreProcess func(img image.Image, targetWidth uint, targetHeight uint) ([]float32, error)

+ func NewYoloNAS(modelName string, modelVersion string) Model

+ func NewYoloV8(modelName string, modelVersion string) Model

+ type ModelType string

+ const ModelTypeYoloNAS

+ const ModelTypeYoloV8

+ type YoloNAS struct

+ func (y *YoloNAS) GetConfig() YoloTritonConfig

+ func (y *YoloNAS) PostProcess(rawOutputContents [][]byte) ([]Box, error)

+ func (y *YoloNAS) PreProcess(img image.Image, targetWidth uint, targetHeight uint) ([]float32, error)

+ type YoloV8 struct

+ func (y *YoloV8) GetConfig() YoloTritonConfig

+ func (y *YoloV8) PostProcess(rawOutputContents [][]byte) ([]Box, error)

+ func (y *YoloV8) PreProcess(img image.Image, targetWidth uint, targetHeight uint) ([]float32, error)

v0.1.1

Sep 3, 2023

v0.1.0

Sep 3, 2023

Changes in this version

+ func DrawBoundingBoxes(img image.Image, boxes []Box, lineWidth int) (image.Image, error)

+ func LoadImage(imagePath string) (image.Image, error)

+ func ModelInferRequest(client triton.GRPCInferenceServiceClient, ...) (*triton.ModelInferResponse, error)

+ func ModelMetadataRequest(client triton.GRPCInferenceServiceClient, modelName string, ...) (*triton.ModelMetadataResponse, error)

+ func SaveImage(img image.Image, filename string) error

+ func ServerLiveRequest(client triton.GRPCInferenceServiceClient) (*triton.ServerLiveResponse, error)

+ func ServerReadyRequest(client triton.GRPCInferenceServiceClient) (*triton.ServerReadyResponse, error)

+ type Box struct

+ Class string

+ Probability float64

+ X1 float64

+ X2 float64

+ Y1 float64

+ Y2 float64

+ type YoloTriton struct

+ func New(url string, cfg YoloTritonConfig) (*YoloTriton, error)

+ func (y *YoloTriton) Close() error

+ func (y *YoloTriton) Infer(img image.Image) ([]Box, error)

+ type YoloTritonConfig struct

+ BatchSize int

+ Height int

+ MaxIOU float64

+ MinProbability float64

+ ModelName string

+ ModelVersion string

+ NumChannels int

+ NumObjects int

+ Width int