I have run into the following error when running s/cre

It works like a charm with <a class="issue-link js-issue-link" data-error-text="Failed

RuntimeError: Tensors must have same number of dimensions: got 4 and 5 about dss HOT 3 CLOSED

yifita commented on June 20, 2024

RuntimeError: Tensors must have same number of dimensions: got 4 and 5

from dss.

Comments (3)

pbonazzi commented on June 20, 2024 2

Well done , I have already managed to solve it on my own , by tracing back the problem in pytorch3d library .

It had to do with the renderer.shader function , in your script I have done the following

            # add these imports
            from pytorch3d.renderer.blending import BlendParams, hard_rgb_blend
            from pytorch3d.renderer.mesh.shading import flat_shading

            # replace these lines (206)
            # images = renderer.shader(fragments, meshes_batch, lights=lights, cameras=cams)
            texels = meshes_batch.sample_textures(fragments)
            colors = flat_shading(meshes=meshes_batch, fragments=fragments, lights=lights, cameras=cams, materials=renderer.shader.materials, texels=texels)
            images = hard_rgb_blend(colors[0], fragments, BlendParams())

from dss.

pbonazzi commented on June 20, 2024 1

It works like a charm with #21

from dss.

lxxue commented on June 20, 2024

Hi, this is probably a pytorch3d version problem. We were working with an old version, and the new version is incompatible. I update the installation part in the fix/install branch. Could you check it out and create the environment again to see if you still have an issue? Things work fine on my side with this new installation.

from dss.

Related Issues (20)

Recommend Projects