When viewing the dashboard I am logging some images with the model mask prediction on them. As seen in the screen shot I can view the images and masks 1 at a time when shown in the default layout. However when I click the image to full screen it, the mask disappears and it only shows the image. When I click the image to then download the mask appears in the downloaded file. How do I make it so when I click the image in the wandb dashboard I still see the image in full screen view. Here is the code I use:
"prediction": wandb.Image(
input_image,
masks={
"predictions": {
"mask_data": instance_mask_resized,
"class_labels": class_labels
}
},
caption=f"Instance Segmentation - Iter {self.iter}"
),
"iteration": self.iter
})