feat: adjusted sample_nums (#1768)

Summary:
adjusted sample_nums to match the number of columns in the image grid. It originally produced image grid with 5 axes but only 3 images and after this fix, the block would work as intended.

Pull Request resolved: https://github.com/facebookresearch/pytorch3d/pull/1768

Reviewed By: MichaelRamamonjisoa

Differential Revision: D55632872

Pulled By: bottler

fbshipit-source-id: 44d633a8068076889e49d49b8a7910dba0db37a7
This commit is contained in:
Ashim Dahal 2024-04-02 06:02:48 -07:00 committed by Facebook GitHub Bot
parent 31e3488a51
commit 128be02fc0

View File

@ -411,7 +411,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"random_model_images = shapenet_dataset.render(\n", "random_model_images = shapenet_dataset.render(\n",
" sample_nums=[3],\n", " sample_nums=[5],\n",
" device=device,\n", " device=device,\n",
" cameras=cameras,\n", " cameras=cameras,\n",
" raster_settings=raster_settings,\n", " raster_settings=raster_settings,\n",