Model.GetClass
has been removed in favor of newYoloTritonConfig.Classes
- Predefined YOLO model init function now receive
YoloTritonConfig
as argument.
- Add
-p <MinProbability>
,-o <MaxIOU>
flag for sample script.
YoloTritonConfig.NumChannels
has been renamed toYoloTritonConfig.NumClasses
.YoloTritonConfig.BatchSize
has been removed.Model.PreProcess
interface has been changed toPreProcess(img image.Image, targetWidth uint, targetHeight uint) (*triton.InferTensorContents, error)
- Add support for YOLO-NAS INT8 inference.
- Add benchmark script.
- Model metadata request will only get once instead everytime before model inference request.
- Add support for YOLO-NAS inference.
- Fix go pkg publish issue.
Initial release of yolotriton.
- Support for YOLOv8 inference.