search

Banconxuan / RTM3D

The official PyTorch Implementation of RTM3D and KM3D for Monocular 3D Object Detection
MIT License
453 stars 85 forks source link

根据dim, keypoints, orientation等计算depth的这部分模块，部署推理时该怎么实现呢？ #51

Open likegogogo opened 2 years ago

likegogogo commented 2 years ago

image

训练时depth的计算目测是耦合在Position_loss 中的。 但部署推理时，不需要算loss，只需要算出depth即可。 请问下，怎么推理实现怎么搞比较好呢？ 我看depth计算这一块挺复杂的。 直接用c++手写？还是把Position_loss 的forward部分封装成torch script 独立运行?

有建议吗？

LuckySunnye commented 2 years ago

Hi 请问你有思路嘛？ 我也要做线上部署的任务。可以交流下吗？