【技术实现步骤摘要】
节点排布方式确定方法及装置、电子设备和存储介质
[0001]本公开涉及计算机
，尤其涉及一种节点排布方式确定方法及装置、电子设备和存储介质。
技术介绍
[0002]在图像处理领域中，被处理图像具有多种不同的数据排布方式。不同的图像处理算子在处理不同排布方式时的性能具有一定的差异，例如部分算子在处理NHWC排布方式的图像数据性能良好，而部分算子在处理NCHW排布方式的图像数据性能良好。因此，为了得到更高效的图像处理效率，需要在图像处理时确定参与图像处理的各算子的排布方式。
技术实现思路
[0003]本公开提出了一种节点排布方式确定方法及装置、电子设备和存储介质，旨在通过确定一种包括多个具有对应执行顺序的图像处理节点对应图像排布方式的最优策略，提高图像处理的效率。
[0004]根据本公开的第一方面，提供了一种节点排布方式确定方法，所述方法包括：
[0005]确定多个第一节点，各所述第一节点为用于根据对应执行顺序处理同一图像数据的算子，具有至少一种对应的图像排布方式，所述图像排布方式用于表征图...
【技术保护点】
【技术特征摘要】
1.一种节点排布方式确定方法，其特征在于，所述方法包括：确定多个第一节点，各所述第一节点为用于根据对应执行顺序处理同一图像数据的算子，具有至少一种对应的图像排布方式，所述图像排布方式用于表征图像数据中各像素值的存储顺序；在每两个相邻的第一节点之间插入第二节点，各所述第二节点用于对输入的图像数据进行排布方式转换；根据各所述第一节点对应的至少一个图像排布方式拆分各所述第一节点和第二节点，得到多个具有对应执行顺序的第一子节点和第二子节点，各所述第一子节点和第二子节点均具有一种对应的图像排布方式；根据各所述第一子节点和第二子节点对应的执行时间确定目标处理策略，所述执行时间用于表征对应第一子节点的图像处理时间或对应第二子节点的排布方式转换时间，所述目标处理策略中包括各所述第一节点对应的目标排布方式。2.根据权利要求1所述的方法，其特征在于，所述根据各所述第一节点对应的至少一个图像排布方式拆分各所述第一节点和第二节点，得到多个具有对应执行顺序的第一子节点和第二子节点包括：根据各所述第一节点对应的至少一个图像排布方式拆分所述第一节点和第二节点，得到各所述第一节点对应的多个第一子节点，以及各所述第二节点对应的多个第二子节点；确定各所述第一子节点和各所述第二子节点的执行顺序。3.根据权利要求2所述的方法，其特征在于，所述根据各所述第一节点对应的至少一个图像排布方式拆分所述第一节点和第二节点，得到各所述第一节点对应的多个第一子节点，以及各所述第二节点对应的多个第二子节点包括：根据对应的图像排布方式拆分各所述第一节点，得到分别与所述第一节点对应各图像排布方式对应的多个第一子节点；确定位于各所述第二节点下一执行顺序的第一节点对应的图像排布方式；拆分各所述第二节点，得到分别对应于下一执行顺序的第一节点对应各图像排布方式的多个第二子节点。4.根据权利要求2或3所述的方法，其特征在于，所述确定各所述第一子节点和各所述第二子节点的执行顺序包括：确定各所述第一子节点的执行顺序与对应的第一节点相同；确定各所述第二子节点的执行顺序与对应的第二节点相同，其中，各所述第一子节点对应的图像排布方式与前继子节点对应的图像排布方式相同。5.根据权利要求1
‑
4中任意一项所述的方法，其特征在于，所述根据各所述第一子节点和第二子节点对应的执行时间确定目标处理策略包括：根据预设的数据执行方式和各所述第一子节点和第二子节点对应的执行时间确定最短执行时间；根据所述最短执行时间确定所述目标处理策略。6.根据权利要求5所述的方法，其特征在于，所述根据预设数据执行方式和各所述第一子节点和第二子节点对应的执行时间确定最短执行时间包括：根据预设数据执行方式确定对应的最短时间计算规则；
根据所述最短时间计算规则由各输入子节点开始，依次向后计算最短处理时间直到各输出子节点，得到多个网络执行时间，所述输入子节点和所述输出子节点分别为根据对应执行顺序第一次和最后一次进行图像数据处理的第一子节点；确定各所述网络执行时间的最小值为最短执行时间。7.根据权利...
【专利技术属性】
技术研发人员：许思，李天健，葛佳烨，
申请(专利权)人：北京市商汤科技开发有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。