Skip to content

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;指定到具体路径和文件名,将以指定路径和文件名命名离线网络模型;不指定该参数,将以网络模型文件路径储存离线网络模型。