通过训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,我们需要通过pdseg/export_model.py
来导出该模型。
该脚本的使用方法和train.py/eval.py/vis.py
完全一样。
FLAG | 用途 | 默认值 | 备注 |
---|---|---|---|
--cfg | 配置文件路径 | None |
我们使用训练/评估/可视化一节中训练得到的模型进行试用,命令如下
python pdseg/export_model.py --cfg configs/unet_optic.yaml TEST.TEST_MODEL ./saved_model/unet_optic/final
预测模型会导出到freeze_model
目录,用于C++
或者Python
预测的模型配置会导出到freeze_model/deploy.yaml
下