In the step function in triain.py, you return the following things from the network
cls_out, cls_fg, masks, mask_logits, pseudo_gt, loss_mask = self.enc(image, image_raw, gt_labels)
What do these things represent:
- cls_out
- cls_fg
- masks
- mask_logits
- pseudo_gt
- moss_mask