#onnx #machine-learning #cli-command #flops

nightly bin+lib steelix

您的ONNX模型分析一站式CLI。具有图可视化、FLOP计数、内存指标等功能!

1个不稳定版本

0.1.0 2022年10月15日

#711机器学习

MIT 许可证

105KB
2.5K SLoC

您的ONNX模型分析一站式CLI。

具有图可视化、FLOP计数、内存指标等功能!

GitHub crates.io GitHub release

⚡️ 快速开始

首先,下载并安装DOT。 可以通过 cargo 来安装。

cargo install steelix

MacOS用户也可以通过HomeBrew安装。

brew tap FL33TW00D/steelix
brew install steelix

⚙️ 命令与选项

Steelix有2个核心功能 - 模型摘要和模型可视化。

摘要

CLI命令用于总结您模型的核心方面。

steelix summary --model-path ./my-model.onnx
选项 描述 类型 默认 必需?
--model-path 您模型所在的路径。 bool false

plot

CLI命令将您的模型作为SVG文件绘制 - 包含推断的形状。

steelix plot --model-path ./my-model.onnx --open 
选项 描述 类型 默认 必需?
--model-path 您模型所在的路径。 string None
--output-path 您的SVG将被保存的路径。 string ./model.svg
--open 在生成后,在浏览器中打开SVG。 boolean false
--disable-shapes 禁用形状推断。 boolean false

支持的操作符(参考 ONNX IR

操作符 实现
Abs
Acos
Acosh
Add
And
ArgMax
ArgMin
Asin
Asinh
Atan
Atanh
AveragePool
BatchNormalization
BitShift
Cast
Ceil
Clip
Compress
Concat
ConcatFromSequence
Constant
ConstantOfShape
Conv
ConvInteger
ConvTranspose
Cos
Cosh
CumSum
DepthToSpace
DequantizeLinear
Det
Div
Dropout
Einsum
Elu
Equal
Erf
Exp
Expand
EyeLike
Flatten
Floor
GRU
Gather
GatherElements
GatherND
Gemm
GlobalAveragePool
GlobalLpPool
GlobalMaxPool
Greater
GridSample
HardSigmoid
Hardmax
Identity
If
InstanceNormalization
IsInf
IsNaN
LRN
LSTM
LeakyRelu
Less
Log
Loop
LpNormalization
LpPool
MatMul
MatMulInteger
Max
MaxPool
MaxRoiPool
MaxUnpool
Mean
Min
Mod
Mul
Multinomial
Neg
NonMaxSuppression
NonZero
Not
OneHot
Optional
OptionalGetElement
OptionalHasElement
Or
PRelu
Pad ✅ (mode=constant, pads≥0)
Pow
QLinearConv
QLinearMatMul
QuantizeLinear
RNN
RandomNormal
RandomNormalLike
RandomUniform
RandomUniformLike
Reciprocal
ReduceL1
ReduceL2
ReduceLogSum
ReduceLogSumExp
ReduceMax
ReduceMean
ReduceMin
ReduceProd
ReduceSum
ReduceSumSquare
Relu
Reshape
Resize
ReverseSequence
RoiAlign
Round
Scan
Scatter (已弃用)
ScatterElements
ScatterND
Selu
SequenceAt
SequenceConstruct
SequenceEmpty
SequenceErase
SequenceInsert
SequenceLength
Shape
Shrink
Sigmoid
Sign
Sin
Sinh
Size
Slice
Softplus
Softsign
SpaceToDepth
Split
SplitToSequence
Sqrt
Squeeze
StringNormalizer
Sub
Sum
Tan
Tanh
TfIdfVectorizer
ThresholdedRelu
Tile
TopK
Transpose
Trilu
Unique
Unsqueeze
Upsample (已弃用)
Where
Xor
Function
Bernoulli
CastLike
Celu
DynamicQuantizeLinear
GreaterOrEqual
HardSwish
LessOrEqual
LogSoftmax
MeanVarianceNormalization
NegativeLogLikelihoodLoss
Range
Softmax
SoftmaxCrossEntropyLoss

Credit

本项目中大部分好主意/代码都受到了 极大 的启发,例如来自 tractwonnxnetron

依赖关系

~12–23MB
~353K SLoC