Yu Sun
Yu Sun
建议您正确加载pretrain来训练,就是使用[这个函数](https://github.com/Arthur151/ROMP/blob/87e80901acbe8dd40bf975a5da010a0cd3357b35/romp/lib/models/modelv1.py#L34).同时设置configs里的`fine_tune: False, eval: False, model_path: None`。 确认正确加载pretrain: log里不出现`INFO:root:missing parameters of layers:['backbone......`。 Q1:为了避免显存溢出,可以在configs里增加一个设置,`max_supervise_num: 128`,这个是控制每个batch里学习人体形态的数量,避免有的时候一个batch里加载的都是密集人群的图片,导致一个batch里的人数太多。您可以根据您的显存酌情减少这个数量。 Q2:如果您想使用之前训练好的模型,继续进行训练的话,在configs里设置好,log里出现的中间模型的地址(在您log的第50453行),例如: ``` fine_tune: True model_path: /data/wxl/romp/checkpoints/hrnet_cm64_myv2_train_pretrained_hrnet_on_gpu0,1_val/100_104.8944_65.8962_hrnet_cm64_myv2_train_pretrained_hrnet.pkl ``` 关于继续训练,如果您没有使用pretrain的话,还是需要始终保持学习率为5e-5比较合适,您可以看到,您训到后面学习率降到5e-7了,已经学不动了,要做这个调整,可以设置configs里的`adjust_lr_factor: 1` 。 epoch数也可以设置为200。
Hi, Vivian Yes, I think you have find the correct answer. Setting the mode to video would allow users to use temporal optimization functions. To use ROMP in 'webcam' mode...
Hi, @MoyGcc Thanks for you kind word! You can use this function to achieve this. https://github.com/Arthur151/ROMP/blob/91dac0172c4dc0685b97f96eda9a3a53c626da47/simple_romp/romp/utils.py#L331 It take the estimated 3D joints, the 2D joints pj2d_org, image size, and the...
That's clever. Glad to hear that.
@Andyen512 No, The image size should be the original size on input image, not on the resize BEV's input map. It is fine to directly use the camera intrinsic in...
@Andyen512 That focal length (23043.9) / image center coords (940.19, 539.23) is just for training humannerf in their camera extrinsic matrix. To inference on you own video, you can re-calculate...
Oh, please at least include the `image,mesh` in the `show_items `. The code is to draw the center confidence on the rendered mesh image. Without `image,mesh` in show_items, it has...
感谢bug report,这是因为在bev和romp里对同一个东西的命名不同,bev里叫top_score,romp里叫centers_conf。 所以如果想直接跑,不修正这个bug的话,需要用bev来跑结果。就是把romp换成bev,其他命令都一样。 要是还是想用romp,就是更新一下romp。我刚推了一版统一了这个名称的版本到GitHub,要用新的, 先在ROMP里git pull更新一下,当然您也可以照着[这个commit](https://github.com/Arthur151/ROMP/commit/7b4734270672e602f4fc5659d37479478e72b78b)手动修改一下 然后到simple-romp里重新安装一下 ``` cd ROMP/simple-romp python setup.py install ```
一欧元滤波施加在全局旋转,会导致异常的突然翻转。目前还没有找到导致这一异常翻转的原因,因此没有施加到全局旋转上。
不是,ROMP_HRNet32_V1.pkl是从0训出来的。pretrain_hrnet.pkl是我为了让大家可以快速复现,不用像我一样每次训特别久,特别制作的。