Skip to main content
. 2020 Dec 23;21(1):43. doi: 10.3390/s21010043
Algorithm 1 The learning process of MAUNet
Input: The training images Itrain, max-epochs E=12, the number of ItrainNtrain, The testing image Itest and the
 groundtruth labels G
Output: The output prediction G, and its performance results GDice, GIoU, GRoC and Gtime
 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 i=1; iE; i++do
  Get the data batch from Itrain
  for j=1; jNtrain; j++ do
   Compute IoU loss function LIoU;
   Compute Dice loss LDice;
   Compute Focal loss LFocal;
   Train MAUNet by optimizing loss LMAUNet and update the weights and parameters;
  end for
 end for
 Testing Stage:
 Feed Itest into the well-trained MAUNet and then output the prediction segmentation G;
 Compute the performance results GDice (Equation (14)), GIoU (Equation (15)), GRoC (Equation (16)) and running time Gtime
 returnG, GDice, GIoU, GRoC and Gtime.