4. Compiler
4. COMPILER¶
Compiler工具位置在 SGS_IPU_SDK/Scripts/calibrator/compiler.py 。 该工具是将SigmaStar定点网络模型转换为SigmaStar离线网络模型。 在SGS_IPU_SDK ⽬录下运⾏以下脚本,输出Library的路径(已经做过该步骤可忽略):
cd ~/SGS_IPU_SDK
source cfg_env.sh
python3 compiler.py \
-m ~/SGS_Models/tensorflow/ssd_mobilenet_v1/ssd_mobilenet_fixed.sim
-m
, --model
: 网络模型文件路径。
可选参数:
-c
, --category
: 模型的类别,主要有Classification / Detection / Unknown。
Classification
: 模型有1个输出,会根据输出排序输出分数由高至低的前5个。Detection
: 模型有4个输出,会根据输出转换为输入图片的bbox位置以及类别。只支持SigmaStar后处理算子[SigmaStar后处理模块]。其他后处理请使用Unknown。Unknown
: 模型输出不属于上述两种类型,会输出全部的Tensor数值。定点网络模型转换到离线网络模型时默认为Unknown。
-o
, --output
: 模型输出路径。指定离线网络模型输出数据位置:指定到文件夹,将自动以网络模型文件前缀命名,后接sgsimg.img;指定到具体路径和文件名,将以指定路径和文件名命名离线网络模型;不指定该参数,将以网络模型文件路径储存离线网络模型。