soft_argmax

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

soft_argmax相关推荐

resnet50代码_13、SOTA论文实践-学习ResNet（80行代码搞定残差backbone网络）
0.论文 Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB I ...
姿态估计之2D人体姿态估计（1）（仅供个人参考）
参考自顶向下的 2D 人体姿态估计 - 知乎人体姿态估计(Human Pose Estimation)经典方法整理 - 知乎 2D人体姿态估计浅析 - 知乎人体姿态估计中回归出了heatmap如 ...
softargmax(soft-argmax)
softargmax可以作为argmax的近似. 因为argmax是不可导的,而softargmax是可导的或者写作思想就是使得softmax之后的结果,实现最大的那个很大,无限接近于1,而其他的 ...
2022-9-19周报
一.阅读:关于视觉识别领域发展的个人观点 1.CV和NLP的三大差异和挑战 CV和NLP的差异.CV三大挑战及应对方法语义稀疏性:解决方案为构建高效计算模型(神经网络)和 ...

soft_argmax

soft_argmax相关推荐

最新文章

热门文章