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

[likegogogo]

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

有建议吗？

[LuckySunnye]

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