keesschollaart81 / yolov5-net

YOLOv5 object detection with C#, ML.NET, ONNX

Yolov5Net

YOLOv5 object detection with ML.NET, ONNX

Installation

Run this line from Package Manager Console:

Install-Package Yolov5Net -Version 1.0.2

Usage

Yolov5Net contains two COCO pre-defined models: YoloCocoP5Model, YoloCocoP6Model.

If you have custom trained model, then inherit from YoloModel and override all the required properties and methods. See YoloCocoP5Model or YoloCocoP6Model implementation to get know how to wrap your own model.

var image = Image.FromFile("assets/test.jpg");

byte[] weights = File.ReadAllBytes("assets/weights/yolov5s6.onnx");

var scorer = new YoloScorer<YoloCocoP6Model>();

List<YoloPrediction> predictions = scorer.Predict(image, weights);

using (var graphics = Graphics.FromImage(image))
{
	foreach (var prediction in predictions) // iterate predictions to draw results
	{
		double score = Math.Round(prediction.Score, 2);

		graphics.DrawRectangles(new Pen(prediction.Label.Color, 1),
			new[] { prediction.Rectangle });

		var (x, y) = (prediction.Rectangle.X - 3, prediction.Rectangle.Y - 23);

		graphics.DrawString($"{prediction.Label.Name} ({score})",
			new Font("Arial", 16, GraphicsUnit.Pixel), new SolidBrush(prediction.Label.Color),
			new PointF(x, y));
	}

	image.Save("assets/result.jpg");
}