Table 1.

Comparison of results between weight pruning-UNet and other models

Model	Training loss	Training accuracy	Mean IOU
UNet	0.5601	97.87	0.435
UNet (depth-wise + BN)	0.4439	93.62	0.362
WP-UNet (network pruning + depth-wise + BN)	0.066	98.43	0.428

BN – Batch normalization; WP – Weight pruning; IOU – Intersection over union