423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-24 16:20:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

tutorial fixes from #336. Wheels with cuda10.1.

Browse Source 
Summary:
Add a document to explain how to run the tutorials.
Fix API of TexturesVertex in fit_textured_mesh.
Prepare cuda 10.1 wheels (not 10.2) for linux to be available on pypi - this matches what colab has.
Change the tutorials to use these new wheels.

Reviewed By: gkioxari

Differential Revision: D23324479

fbshipit-source-id: 60e92a3f46a2d878f811b7703638f8d1dae143d9
This commit is contained in:
Jeremy Reizenstein
2020-08-25 13:09:43 -07:00
committed by Facebook GitHub Bot
parent 909dc83505
commit 32484500be
11 changed files with 63 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/tutorials/fit_textured_mesh.ipynb
View File

@@ -60,7 +60,11 @@
"outputs": [],
"source": [
"!pip install torch torchvision\n",
"!pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
"import sys\n",
"if torch.__version__=='1.6.0+cu101' and sys.platform.startswith('linux'):\n",
" !pip install pytorch3d\n",
"else:\n",
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
]
},
{
@@ -781,7 +785,7 @@
" new_src_mesh = src_mesh.offset_verts(deform_verts)\n",
" \n",
" # Add per vertex colors to texture the mesh\n",
" new_src_mesh.textures = TexturesVertex(verts_rgb=sphere_verts_rgb) \n",
" new_src_mesh.textures = TexturesVertex(verts_features=sphere_verts_rgb) \n",
" \n",
" # Losses to smooth /regularize the mesh shape\n",
" loss = {k: torch.tensor(0.0, device=device) for k in losses}\n",