VITA-Group / FasterSeg

[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang
MIT License
526 stars 107 forks source link

Head: Paper vs Code #16

Closed theevann closed 4 years ago

theevann commented 4 years ago

Hello !

In the paper, the heads are fairly clearly defined with the picture.

In your code, I see that and I don't exactly understand what this is for :

image

chenwydj commented 4 years ago

Hi @theevann ! Thank you for your interest!

In https://github.com/TAMU-VITA/FasterSeg/blob/master/search/model_search.py: Line 338 ~ 344: progressive feature aggregation (orange dashed rectangular in paper). Line 346 ~ 350: final projection layer for pixel-wise classification.