deploy: 0a200cd

nerfstudio-project · Sep 30, 2024 · 846bbeb · 846bbeb
1 parent 9ecd60e
commit 846bbeb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/main/.doctrees/environment.pickle b/main/.doctrees/environment.pickle
diff --git a/main/_modules/gsplat/rendering.html b/main/_modules/gsplat/rendering.html
@@ -1626,7 +1626,9 @@ <h1>Source code for gsplat.rendering</h1><div class="highlight"><pre>
         <span class="s2">&quot;gradient_2dgs&quot;</span><span class="p">:</span> <span class="n">densify</span><span class="p">,</span>  <span class="c1"># This holds the gradient used for densification for 2dgs</span>
     <span class="p">}</span>
 
-    <span class="n">render_normals</span> <span class="o">=</span> <span class="n">render_normals</span> <span class="o">@</span> <span class="n">torch</span><span class="o">.</span><span class="n">linalg</span><span class="o">.</span><span class="n">inv</span><span class="p">(</span><span class="n">viewmats</span><span class="p">)[</span><span class="mi">0</span><span class="p">,</span> <span class="p">:</span><span class="mi">3</span><span class="p">,</span> <span class="p">:</span><span class="mi">3</span><span class="p">]</span><span class="o">.</span><span class="n">T</span>
+    <span class="n">render_normals</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">einsum</span><span class="p">(</span>
+        <span class="s2">&quot;...ij,...hwj-&gt;...hwi&quot;</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">linalg</span><span class="o">.</span><span class="n">inv</span><span class="p">(</span><span class="n">viewmats</span><span class="p">)[</span><span class="o">...</span><span class="p">,</span> <span class="p">:</span><span class="mi">3</span><span class="p">,</span> <span class="p">:</span><span class="mi">3</span><span class="p">],</span> <span class="n">render_normals</span>
+    <span class="p">)</span>
 
     <span class="k">return</span> <span class="p">(</span>
         <span class="n">render_colors</span><span class="p">,</span>