search

edvardHua / PoseEstimationForMobile

:dancer: Real-time single person pose estimation for Android and iOS.
Apache License 2.0
1k stars 268 forks source link

有关数据集问题 #32

Open accountcwd opened 6 years ago

accountcwd commented 6 years ago

@edvardHua 作者你好，想问一下你的2.2W张单人图片是怎么筛选出来的？我从AIchallenger数据集中筛选只有单人的图片却能找出7W张。我是将BBOX裁出来当作训练数据进行训练的，但是使用你的网络结构，无论是全集27W张，还是在只有一个人的图片裁出来的数据集（7W），训练出来的效果跟你的比还是有很大的差距。你的pckh：71.57，我的最多只有50。希望能解答我的疑问，谢谢～

edvardHua commented 6 years ago

这个项目假设输入的图片是单人的，且人占整个图片面积较大的前提下来筛选数据集的。

我在训练时没有用到 bbox，直接是图片 resize 输入进去训练，如果你是先用 bbox 裁剪图片后再训练的话，那么输入的比例改成人的比例或许会更好。

希望我的回答能帮到你。

accountcwd commented 6 years ago

的确我考虑过比例的问题，以128x256作为输入训练出来的效果还没有192x192的好。我在想是不是裁出bbox训练出的模型对背景辨别低的问题，你了解过其他state of art 的方法数据处理是整张图训练还是裁剪过的吗？