Skip to content

Instantly share code, notes, and snippets.

@olokobayusuf

olokobayusuf/MobileNet.cs Secret

Created September 28, 2021 16:32
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save olokobayusuf/ffabc59761bda48025d448164672bdaa to your computer and use it in GitHub Desktop.
Save olokobayusuf/ffabc59761bda48025d448164672bdaa to your computer and use it in GitHub Desktop.
Download ZIP
Raw
MobileNet.cs
// Create an array feature
var logits = new MLArrayFeature<float>(...); // shape: (1, 1000)
// Perform argmax
var result = logits.ArgMax(dim: 1).Item;
var resultLabel = labels[result];
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment