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

Update subplot arrangement to support non-uniform grids

Browse Source 
Summary: Previously, grids where the columns don't divide the number of plots evenly would error. Now, there'll just be a sparse last row.

Reviewed By: bottler

Differential Revision: D25069236

fbshipit-source-id: 9d2fd62f3d39bfebc07ce0a41718621fa69d6057
This commit is contained in:
Amitav Baruah 2020-11-18 15:45:42 -08:00 committed by Facebook GitHub Bot
parent 5fb63b4520
commit 6c2fc685de
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pytorch3d/vis/plotly_vis.py
View File

@ -692,10 +692,9 @@ def _gen_fig_with_subplots(batch_size: int, ncols: int, subplot_titles: List[str
Returns: Returns:
Plotly figure with ncols subplots per row, and batch_size subplots. Plotly figure with ncols subplots per row, and batch_size subplots.
""" """
if batch_size % ncols != 0:
msg = "ncols is invalid for the given mesh batch size."
warnings.warn(msg)
fig_rows = batch_size // ncols fig_rows = batch_size // ncols
if batch_size % ncols != 0:
fig_rows += 1 # allow for non-uniform rows
fig_cols = ncols fig_cols = ncols
fig_type = [{"type": "scene"}] fig_type = [{"type": "scene"}]
specs = [fig_type * fig_cols] * fig_rows specs = [fig_type * fig_cols] * fig_rows