| Algorithm 2 The learning process of SASAPD |
|
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 |
| for ; ; do |
| Get the data batch from |
| Compute loss function (Equation (7)) |
| Each instance is assigned to the pyramid level which has the minimal loss |
| Train SASAPD by optimizing loss (Equation (10)) where , and update the weights and parameters; |
| end for |
| end for |
| for ; ; do |
| for ; ; do |
| Get the data batch from |
| Compute loss function (Equation (7)) |
| Train SASAPD by optimizing loss (Equation (10)), and update the weights and parameters; |
| end for |
| end for |
| Feed into the well-trained SASAPD and then output the prediction segmentation ; |
| Compute the performance results (Equation (14)), (Equation (17)), (Equation (17)) and running time |
| return , , and . |