python - 如何从 pytorch 输出张量中提取具有置信度得分的关键点？

Question

我正在使用 Pytorch 进行姿势估计的预训练模型： https ://github.com/microsoft/human-pose-estimation.pytorch

我正在使用该模型来检测图像中的人及其 17 个关键点，但我不确定如何将输出张量转换为每个关键点的像素坐标格式，并为每个人提供置信度分数。

这是预测的代码：

model.eval()
img_t = ToTensor()(img).unsqueeze(0)
out = model(img_t)
out.shape

输出形状为 [1, 16, 64, 56]。我有点失落。任何帮助，将不胜感激。

标签： pythonpytorchpose-estimation