EPro-PnP是否可以单独拿出来结合其他模型在KITTI数据集上训练？

tjiiv-cprg / EPro-PnP

[CVPR 2022 Oral, Best Student Paper] EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation

https://www.youtube.com/watch?v=TonBodQ6EUU

Apache License 2.0

1.09k stars 104 forks source link

EPro-PnP是否可以单独拿出来结合其他模型在KITTI数据集上训练？ #63

Open MINGYUAN-W opened 1 year ago

MINGYUAN-W commented 1 year ago

您好！很感谢您能分享这么棒的工作的代码？请问EPro-PnP是否可以单独拿出来结合其他模型在KITTI数据集上训练？

Lakonik commented 1 year ago

Yes of course. The FCOS3D detector is a bit out of date now AFAIK, so I think there could be a lot more potential if you plug EPro-PnP into a more advanced model

MINGYUAN-W commented 1 year ago

Hello! I have another question, how long does it take for EPro-PnP-Det to train once on the nuScenes dataset?

Lakonik commented 1 year ago

About two days (for 12 epochs) using 2 RTX 3090 GPUs