VainF / DeepLabV3Plus-Pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Instance segmentation

fortminors opened this issue · comments

Hello!

How would I extend this implementation for instance segmentation task? I would like to be able to differentiate between different objects of the same class and feed separate masks during training process so that the model predicts distinct objects. Thanks in advance!

Hi @fortminors, you can implement a new head like this for instance segmentation.