用pytorch-pose的lsp训练时遇到的问题
项目地址:https://github.com/bearpaw/pytorch-pose
作者提供的标注是lspet数据集的而不是lsp的,所以一定要下载正确的数据集。
但是虽然数据集找对了,但是作者的标注又有问题了。在部分图片中,第7、8个关节的坐标是在人体之外的(因为lspet就没有这两个标注,作者为了让lsp跟mpii的关节对应起来才加了这两个关节,虽然不知道作者怎么弄的把大部分图片标注的挺准的,但是在部分图片上这两个关节的位置有点放飞自我),在训练的时候不把这两个关节计入损失倒还好,关键是这两个关节被记为可见并且拿去计算人体中心点了,这就很坑了。看下面这个图,空心圆是作者计算的人体中心,实心圆是我把7、8两个关节去掉后重新计算的人体中心。为什么说坑呢,因为作者在训练的时候是以这个点为中心裁剪图像的,如果这个点压根就不在人体上,那么裁下来的图像可能只包含部分人体。