Semantic segmentation is a fundamental and challenging problem in computer vision. Recent studies attempt to integrate feature information of different depths to improve the performance of segmentation tasks, and a few of them enhance the features before fusion. However, which areas of the feature should be strengthened and how to strengthen are still inconclusive. Therefore, in this work we propose an Adaptive Feature Enhancement Module (AFEM) that utilizes high-level features to adaptively enhance the key areas of low-level features. Meanwhile, an Adaptive Feature Enhancement Network (AFENet) is designed with AFEM to combine all the enhanced features. The proposed method is validated on representative semantic segmentation datasets, Cityscapes and PASCAL VOC 2012. In particular, 79.5% mIoU on the Cityscapes testing set is achieved without using fine-val data, which is 1.1% higher than the baseline network and the model size is smaller.

The weight of the model is under the directory save_models/cityscapes.

• torch 1.1.0
• torchvision 0.3.0
• numpy 1.17.1
• opencv-python 4.1.1.26
• Pillow 6.1.0

Train:

• python train.py

• python train_for_voc.py

Test:

• python validate.py

• python generate_test_result.py
• python generate_result_for_voc.py

• python test_with_an_image.py

• Note: Only the parameters of args need to be modified in the these files.