huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

vig for segmentation

Yakiw opened this issue · comments

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]

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