| Algorithm 1 The learning process of MAUNet |
|
Input: The training images , max-epochs , the number of , The testing image and the groundtruth labels G |
| Output: The output prediction , and its performance results , , and |
| All the images are preprocessed according to the steps in Section 4.2. |
| Training Stage: |
| Initialize the network weights, learning rate, batch size, and other parameters |
| for ; ; do |
| Get the data batch from |
| for ; ; do |
| Compute IoU loss function ; |
| Compute Dice loss ; |
| Compute Focal loss ; |
| Train MAUNet by optimizing loss and update the weights and parameters; |
| end for |
| end for |
| Testing Stage: |
| Feed into the well-trained MAUNet and then output the prediction segmentation ; |
| Compute the performance results (Equation (14)), (Equation (15)), (Equation (16)) and running time |
| return, , , and . |