got root joint location

2022-12-30 12:57:08 +00:00
parent e5780de5e5
commit 51b6219021
3 changed files with 192 additions and 181 deletions
@@ -198,4 +198,4 @@ class DepthRenderer(nn.Module):
        # import plotly.express as ex
        # ex.imshow(depth.cpu().numpy()).show()

-        return depth, mask, v2i, K
+        return depth, mask, v2i, K, cameras