axon_onnx
axon_onnx copied to clipboard
elixir-nx
Easily convert models between ONNX and Axon
Umbrella issue tracking efforts to import the smallest version https://github.com/meituan/YOLOv6 called `YOLOv6-nano`. - https://github.com/elixir-nx/axon_onnx/pull/38
As agreed I have cleaned up the code of an Helper module, which is a partial port from Python ONNX library. This patch is including three files: - lib/axon_onnx/helper.ex which...
Checklist for serialization support of Axon models. ## Linear - [x] `dense` - [ ] `bilinear` - [ ] `embedding` ## Convolutional - [x] `conv` - [ ] `conv_transpose` -...
ONNX pooling operations require that the padding configuration have less padding on each side than the size of the kernel. This isn't validated currently but should be.
While trying to import [MediaPipe Face Mesh model](https://github.com/google-ai-edge/mediapipe/wiki/MediaPipe-Face-Mesh), I encountered this error: ```elixir ** (ArgumentError) unsupported "PRelu" (axon_onnx 0.4.0) lib/axon_onnx/deserialize.ex:2213: AxonOnnx.Deserialize.recur_nodes/2 (elixir 1.17.1) lib/enum.ex:2531: Enum."-reduce/3-lists^foldl/2-0-"/3 (axon_onnx 0.4.0) lib/axon_onnx/deserialize.ex:43: AxonOnnx.Deserialize.graph_to_axon/2 (axon_onnx...