I’m working on a task to denoise microscopy images using a 2D U-Net. I’m training my network on images taken at different z-levels, and these images have ground truth, which is the mean of the images in z. Therefore, some images share the same ground truth. I’ve separated the images into z-groups to place them in either train, validation, or test sets. I’m computing the SSIM on each batch and averaging the results at each epoch to plot them afterward.