search

nv-tlabs / GSCNN

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)
https://nv-tlabs.github.io/GSCNN/
Other
915 stars 200 forks source link

What are shape of map and _edgemap? #38

Open John1231983 opened 4 years ago

John1231983 commented 4 years ago

Thanks for sharing the great project.

If BxCxHxW is shape of the input image, what is shape of mask and _edgemap?

Thks

XGBoost commented 4 years ago

@John1231983 As segmentation is a dense prediction problem, the shape of the mask is B x Class_number x H x W. City spaces dataset has 19 classes, so the final size of the mask is Bx19xHxW. The _edgemap is binary and only has one channel and the shape of it is Bx1xHxW.