跑demo的时候，测试图片是多人，但是检测结果只有一个人被检测到，这是什么问题呢？

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

http://mvig.org/research/alphapose.html

Other

8.02k stars 1.97k forks source link

Closed KCHENPENGFEI closed 5 years ago

KCHENPENGFEI commented 5 years ago

xiaoweiChen commented 5 years ago

用的也是pytorch 0.4.0么?

这个是最新提交引入的bug，https://github.com/MVIG-SJTU/AlphaPose/issues/237 已经提交了。上交的同学正在修复中。

piaozhx commented 5 years ago

我也遇到类似的问题了, 经过debug发现是预处理时候的问题, 把dataloader.py中的

inps[i] = cropBox(tmp_img, upLeft, bottomRight, opt.inputResH, opt.inputResW)

改为

inps[i] = cropBox(tmp_img.clone(), upLeft, bottomRight, opt.inputResH, opt.inputResW)

就可以正常运行了

xiaoweiChen commented 5 years ago

@piaozhx 你的方式的确能修复这个问题

KCHENPENGFEI commented 5 years ago

@piaozhx 我试试看

KCHENPENGFEI commented 5 years ago

@xiaoweiChen 我用的pytorch1.0

xiaoweiChen commented 5 years ago

@KCHENPENGFEI 目前pytorch分支估计对1.0.0支持的不是很好。方便用webcam和video的demo进行一下测试么？看看在1.0.0下还有没有其他的问题。

KCHENPENGFEI commented 5 years ago

@xiaoweiChen 我测试了，1.0版本可以跑video的demo，并且也是可以检测出多人的，由于我没有显示器，webcam可能测试不了了

xiaoweiChen commented 5 years ago

@KCHENPENGFEI 好的，多谢啦！

KCHENPENGFEI commented 5 years ago

@piaozhx pytorch0.4.0版本按照你说的办法是可以的，但是1.0就无效了。

Fang-Haoshu commented 5 years ago

bug fixed, thanks