diff --git a/pytorch3d/implicitron/models/metrics.py b/pytorch3d/implicitron/models/metrics.py index cc44a518..13387609 100644 --- a/pytorch3d/implicitron/models/metrics.py +++ b/pytorch3d/implicitron/models/metrics.py @@ -362,6 +362,11 @@ class ViewMetrics(ViewMetricsBase): def _rgb_metrics(images, images_pred, masks, masks_pred, masks_crop): assert masks_crop is not None + if images.shape[1] != images_pred.shape[1]: + raise ValueError( + f"Network output's RGB images had {images_pred.shape[1]} " + f"channels. {images.shape[1]} expected." + ) rgb_squared = ((images_pred - images) ** 2).mean(dim=1, keepdim=True) rgb_loss = utils.huber(rgb_squared, scaling=0.03) crop_mass = masks_crop.sum().clamp(1.0) diff --git a/pytorch3d/renderer/implicit/raymarching.py b/pytorch3d/renderer/implicit/raymarching.py index 1f3ca9ba..195b6aa6 100644 --- a/pytorch3d/renderer/implicit/raymarching.py +++ b/pytorch3d/renderer/implicit/raymarching.py @@ -215,6 +215,7 @@ def _check_raymarcher_inputs( if density_1d and rays_densities.shape[-1] != 1: raise ValueError( "The size of the last dimension of rays_densities has to be one." + + f" Got shape {rays_densities.shape}." ) rays_shape = rays_densities.shape[:-1]