Skip to content

Lightweight Neural Network for Semantic Segmentation using Knowledge Distillation (Accepted by AICAS 2022)

License

Notifications You must be signed in to change notification settings

yubin1219/Semantic-Seg-KD

Repository files navigation

Class Attention Transfer for Semantic Segmentation

Accepted by IEEE AICAS 2022 (Oral) [Paper]

We generate the class attention maps (CAM) from the feature maps of each student and the teacher network, and then, the student network is guided to mimic the regions that need to be paid more attention. The overall architecture of Class Attention Transfer is illustrated as below.


Visualization Class Attention Maps

The CAM generated by our method from the feature map of the teacher network shows activated regions corresponding to the categories.


About

Lightweight Neural Network for Semantic Segmentation using Knowledge Distillation (Accepted by AICAS 2022)

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages