Brain Tumor Classification and Segmentation Using Dual-Outputs for U-Net Architecture: O2U-Net.

We propose a modified version of the U-Net architecture for segmenting and classifying brain tumors, introducing another output between down- and up-sampling. Our proposed architecture utilizes two outputs, adding a classification output beside the segmentation output. The central idea is to use fully connected layers to classify each image before applying U-Net's up-sampling operations.