images for debugging TexturesUV

Summary: New methods to directly plot a TexturesUV map with its used points, using PIL and matplotlib.

Reviewed By: gkioxari

Differential Revision: D23782968

fbshipit-source-id: 692970857b5be13a35a3175dc82ac03963a73555

docs/tutorials/render_textured_meshes.ipynb

@@ -93,7 +93,7 @@
     "\n",
     "# Data structures and functions for rendering\n",
     "from pytorch3d.structures import Meshes\n",
-    "from pytorch3d.vis import AxisArgs, plot_batch_individually, plot_scene\n",
+    "from pytorch3d.vis import AxisArgs, plot_batch_individually, plot_scene, texturesuv_image_matplotlib\n",
     "from pytorch3d.renderer import (\n",
     "    look_at_view_transform,\n",
     "    FoVPerspectiveCameras, \n",
@@ -236,8 +236,7 @@
     "obj_filename = os.path.join(DATA_DIR, \"cow_mesh/cow.obj\")\n",
     "\n",
     "# Load obj file\n",
-    "mesh = load_objs_as_meshes([obj_filename], device=device)\n",
-    "texture_image=mesh.textures.maps_padded()"
+    "mesh = load_objs_as_meshes([obj_filename], device=device)"
    ]
   },
   {
@@ -265,9 +264,29 @@
    "outputs": [],
    "source": [
     "plt.figure(figsize=(7,7))\n",
+    "texture_image=mesh.textures.maps_padded()\n",
     "plt.imshow(texture_image.squeeze().cpu().numpy())\n",
     "plt.grid(\"off\");\n",
-    "plt.axis('off');"
+    "plt.axis(\"off\");"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "PyTorch3D has a built-in way to view the texture map with matplotlib along with the points on the map corresponding to vertices. There is also a method, texturesuv_image_PIL, to get a similar image which can be saved to a file."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "plt.figure(figsize=(7,7))\n",
+    "texturesuv_image_matplotlib(mesh.textures, subsample=None)\n",
+    "plt.grid(\"off\");\n",
+    "plt.axis(\"off\");"
    ]
   },
   {