Jeff-sjtu / res-loglikelihood-regression

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral
423 stars 44 forks source link

Provide implementation of single person pose estimation visualization. #24

Open larry6799 opened 3 years ago

larry6799 commented 3 years ago

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

主要的修改的地方有:

  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等参数。

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

Jeff-sjtu commented 3 years ago

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

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

再次感谢您的贡献!

larry6799 commented 3 years ago

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

Jeff-sjtu commented 3 years ago

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

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

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

larry6799 commented 3 years ago

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

lbbbbbbbbbb commented 2 years ago

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

larry6799 commented 2 years ago

微信: larry1648637120

---Original--- From: @.> Date: Wed, Nov 17, 2021 13:23 PM To: @.>; Cc: @.**@.>; Subject: Re: [Jeff-sjtu/res-loglikelihood-regression] Provide implementationof single person pose estimation visualization. (PR #24)

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

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.