更新时间:2021-03-18 GMT+08:00
分享

算子开发注意事项

SSD网络中带后处理节点的自定义算子开发注意事项

SSD网络的PriorBox算子与Detection Ouptut算子模型转换时会自动融合,若用户自定义实现了SSD网络中的PriorBox算子,则无法自动融合导致模型转换失败,此时需要用户删除网络模型文件中的Detection Output层算子,然后在Engine编排流程中自定义后处理节点实现Detection Output算子的功能。

其他

  • 请勿修改框架内置算子的实现(路径:ddk/include/inc/custom下的所有文件),否则将可能导致未知风险,例如:系统启动失败、模型无法正常转换等。
  • 客户在开发算子插件时需要对自己的源代码负责,避免植入后门。
分享:

    相关文档

    相关产品