Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Provide implementation of single person pose estimation visualization. #24

Open
wants to merge 4 commits into
base: master
Choose a base branch
from

Conversation

larry6799
Copy link

大佬您好,我是北京理工大学的一名硕士生,对您的工作很感兴趣。但是在学习的过程中发现缺少姿态估计工作应该有的可视化的代码,所以我顺带实现了一下该工作的单人姿态估计可视化的代码,您检查一下看看是否可以接受我的贡献的代码。

主要的修改的地方有:

  1. 修改 readme.md 中关于增加的可视化的描述和演示demo的gif (gif文件放在放图片的assets文件夹)
  2. 增加 scripts/visualize.py, 读取视频进行单人姿态估计可视化,出于尽量解耦 inference 的代码和训练等其他文件,该文件只import 了 建立模型和读取配置的python文件。
  3. 增加运行脚本 scripts/visualize.sh 和 配置文件 configs/256x192_res50_coco_visualize.yaml, 可以自行配置 需要测试的视频文件位置,使用的模型文件,是否使用GPU等参数。

再次给大佬优秀的开源工作点赞!

@jeffffffli
Copy link
Owner

@larry6799 你好,十分感谢你对我们工作的关注以及你提交的pr。在检查代码的过程中我发现一些问题,希望你可以进一步修正一下。

  1. 新增添的config文件中data的路径和其他config文件保持一致。
  2. visualize.py里面有些函数于rlepose/utils/transforms.py内的函数是重复的,可以通过import的形式复用。
  3. 现在visualize.py中是默认输入的视频内只有一个人,且bbox大小设定为整张图片的大小。但实际用的时候这个很难保证。是否可以使用外部传入的文件来提供bbox位置?比如添加一个bbox_list.txt之类的。

再次感谢您的贡献!

@larry6799
Copy link
Author

第1点和第2点我可以保证。
但是对于第三点,我pr的就是单人姿态估计的可视化代码,所以bbox大小设定为整张图片的大小。
而且,我可视化的是一段视频的姿态估计,给一段视频的每一帧通提供bbox_list.txt,是不是更不符合实际情况?

@jeffffffli
Copy link
Owner

第1点和第2点我可以保证。 但是对于第三点,我pr的就是单人姿态估计的可视化代码,所以bbox大小设定为整张图片的大小。 而且,我可视化的是一段视频的姿态估计,给一段视频的每一帧通提供bbox_list.txt,是不是更不符合实际情况?

你好,不是每一帧提供一个txt,是整个视频提供一个txt。可视化的部分可以还是单人的,txt中提供每一个帧这个人所在位置的bbox。这样的话对于单人的可视化效果也会更好,因为即使是单人的视频,人也会在视频中移动。crop出一个tight的bbox送到pose网络也能提供更好效果。

当然,如果第3点您不好实现的话,可以后续交给我们处理。您可以只修改第1第2点。

@larry6799
Copy link
Author

感谢您的理解,我会尽快修改一下第1点和第2点

@lbbbbbbbbbb
Copy link

同学你好,我也是一名硕士生,对于你新添的代码有些问题想要请教,方便留一下联系方式么

@larry6799
Copy link
Author

larry6799 commented Nov 17, 2021 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants