Yolov5Net 1.1.0
dotnet add package Yolov5Net --version 1.1.0
NuGet\Install-Package Yolov5Net -Version 1.1.0
<PackageReference Include="Yolov5Net" Version="1.1.0" />
paket add Yolov5Net --version 1.1.0
#r "nuget: Yolov5Net, 1.1.0"
// Install Yolov5Net as a Cake Addin
#addin nuget:?package=Yolov5Net&version=1.1.0
// Install Yolov5Net as a Cake Tool
#tool nuget:?package=Yolov5Net&version=1.1.0
Yolov5Net
YOLOv5 object detection with ML.NET, ONNX
Installation
Run this line from Package Manager Console:
Install-Package Yolov5Net -Version 1.1.0
For CPU usage run this line from Package Manager Console:
Install-Package Microsoft.ML.OnnxRuntime -Version 1.14.1
For GPU usage run this line from Package Manager Console:
Install-Package Microsoft.ML.OnnxRuntime.Gpu -Version 1.14.1
CPU and GPU packages can't be installed together.
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.
using var image = await Image.LoadAsync<Rgba32>("Assets/test.jpg");
{
using var scorer = new YoloScorer<YoloCocoP5Model>("Assets/Weights/yolov5n.onnx");
{
var predictions = scorer.Predict(image);
var font = new Font(new FontCollection().Add("C:/Windows/Fonts/consola.ttf"), 16);
foreach (var prediction in predictions) // draw predictions
{
var score = Math.Round(prediction.Score, 2);
var (x, y) = (prediction.Rectangle.Left - 3, prediction.Rectangle.Top - 23);
image.Mutate(a => a.DrawPolygon(new Pen(prediction.Label.Color, 1),
new PointF(prediction.Rectangle.Left, prediction.Rectangle.Top),
new PointF(prediction.Rectangle.Right, prediction.Rectangle.Top),
new PointF(prediction.Rectangle.Right, prediction.Rectangle.Bottom),
new PointF(prediction.Rectangle.Left, prediction.Rectangle.Bottom)
));
image.Mutate(a => a.DrawText($"{prediction.Label.Name} ({score})",
font, prediction.Label.Color, new PointF(x, y)));
}
await image.SaveAsync("Assets/result.jpg");
}
}
Product | Versions |
---|---|
.NET | net6.0 net6.0-android net6.0-ios net6.0-maccatalyst net6.0-macos net6.0-tvos net6.0-windows net7.0 net7.0-android net7.0-ios net7.0-maccatalyst net7.0-macos net7.0-tvos net7.0-windows |
-
net6.0
- Microsoft.ML.OnnxRuntime.Managed (>= 1.14.1)
- SixLabors.ImageSharp (>= 3.0.0)
NuGet packages
This package is not used by any NuGet packages.
GitHub repositories
This package is not used by any popular GitHub repositories.