Object detection in Elixir with Axon

Hans Josephsen

@hansihe

Object Detection

Input Tensor

Layer

Nx

Tensor

Tensor

Layer

Nx

Layer

Nx

Layer

Nx

Tensor

Tensor

Output Tensor

YOLO

You Only Look Once

Source: https://github.com/ultralytics/ultralytics

How do we use this in Elixir with Axon?

ONNX

Open Neural Network Exchange

Source: https://towardsdatascience.com/onnx-preventing-framework-lock-in-9a798fb34c92

Source: https://towardsdatascience.com/onnx-preventing-framework-lock-in-9a798fb34c92

  • Export yolov8n to ONNX
  • Import using axon_onnx
  • Run on an image
  • Visualize results

Steps:

20x20

+ 40x40

+ 80x80

= 8400

You still need to understand the model

axon_onnx makes it easy to import and run ML models!

axon_onnx

Nx is really useful for manupulating data!

Nx