如何减少多个3D关键点的丢失

Question

我有一个G = [18000x3]的3D点的基本真值，以及我的网络相同大小的输出O = [18000x3]。

我需要计算一个损失，以便我基本上得到每个3D点之间的距离的平方根，在所有关键点上求和并且标准化超过18000.如何有效地写这个？

Answer 1

只需使用PyTorch提供的向量化操作编写您建议的表达式。在这种情况下

loss = (O - G).pow(2).sum(axis=1).sqrt().mean()

查看pow，sum，sqrt和mean。