soft_argmax

本站寻求有缘人接手，详细了解请联系站长QQ1493399855

https://github.com/Fdevmsy/PyTorch-Soft-Argmax/blob/master/soft-argmax.py

在姿态估计算法中，微软开发的 SimpleBaseline 是精度高而又轻量级的典范，昨天一篇论文Simple and Lightweight Human Pose Estimation，在该架构基础上做了少许改进，取得了更快的速度和更小的模型Size。

论文作者信息：

import torch
import torch.nn as nndef soft_argmax(voxels):"""Arguments: voxel patch in shape (batch_size, channel, H, W, depth)Return: 3D coordinates in shape (batch_size, channel, 3)"""assert voxels.dim()==5# alpha is here to make the largest element r