hszhao / semseg

Semantic Segmentation in Pytorch
MIT License
1.34k stars 244 forks source link

how to get the feature of every instance #85

Open wlufy opened 2 years ago

wlufy commented 2 years ago

The output shape of PSPNet is (batch, class_num, h, w). Now i want to get the feature(one vector) of every class. What shuold i do to get it? Thanks!