NVlabs / RVT

Official Code for RVT-2 and RVT

Home Page:https://robotic-view-transformer-2.github.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Use RVT-2 without pytorch3d?

Walter0807 opened this issue · comments

The readme says that "You can skip this step if you only want to use RVT-2 as it uses our custom Point-Renderer for rendering. PyTorch3D is required for RVT." However, I found that it is still used (for example here). Is it possible to bypass pytorch3d for RVT-2?

Hi @Walter0807,

That's a good point. The augmentation function still uses some utils from PyTorch3D, so the current code bypasses only its rendering functionality. If you want to bypass it altogether, the util functions could be replaced by similar ones from Scipy (more info here). I would be happy to merge any updates if you create one.

Best,
Ankit

Thanks for your reply! I managed to install PyTorch3D for backward compatibility.