Tensorflow:确定预训练的CNN模型的输出步幅

时间:2020-02-05 03:31:40

标签: python tensorflow tensorflow-lite pose-estimation

我已经使用Tensorflow Lite Posenet Model下载并正在执行ML应用程序。该模型的输出是一个热图,它是CNN的新手。

处理输出所需的一条信息是“输出步幅”。用于计算原始图像中找到的关键点的原始坐标。

setText

但是documentation没有指定输出步幅。我可以使用tensorflow中的信息或方法来获取此(任何)预训练模型的输出步幅吗?

  • img的输入形状为:keypointPositions = heatmapPositions * outputStride + offsetVectors
  • 输出形状为:(257,257,3)(17个不同关键点的1 [9x9]热图)
(9,9,17)

1 个答案:

答案 0 :(得分:2)

输出步幅可从以下公式获得:

resolution = ((InputImageSize - 1) / OutputStride) + 1

示例:输入图像,其宽度 225 像素,而输出 跨度为 16 时,输出大小为 15

15 = ((225 - 1) / 16) + 1

对于tflite PoseNet模型:

9 = ((257-1)/ x) + 1 x = 32,因此输出步幅为32

Source

相关问题