search

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.07k stars 708 forks source link

vig for segmentation #232

Closed Yakiw closed 1 year ago

Yakiw commented 1 year ago

Hello, may I ask if vig can be used in semantic segmentation? If so, where should it be modified?

I tried to modify only the prediction layer, but found it difficult to restore the dimensions [1,n_filters,1,1] output before the prediction layer to the original resolution [1,classes,H,W]

iamhankai commented 1 year ago

使用mmsegmentation，替换里面的backbone就行了，https://github.com/open-mmlab/mmsegmentation