code:

gluoncv.model_zoo.get_model(‘fcn_resnet101_coco’,classes=2,ctx=mx.gpu(),pretrained=False)

I download the model with gluoncv. The gluoncv models are 1000 kinds. I want to become the two kinds.How to do?

code:

gluoncv.model_zoo.get_model(‘fcn_resnet101_coco’,classes=2,ctx=mx.gpu(),pretrained=False)

I download the model with gluoncv. The gluoncv models are 1000 kinds. I want to become the two kinds.How to do?

Hi @DarkWings,

have you seen this topic? It has a similar problem and example how to go about it.

1 Like

Thanks

code:fcn_net=gluoncv.model_zoo.get_model(‘fcn_resnet101_coco’,pretrained=False,classes=2)

This is my test code. It will show error the above screenshots.

This works for you?

```
model = gluoncv.model_zoo.FCN(nclass=2,backbone='resnet101',pretrained_base=True)
```

Thanks very much ,This is OK. Thank you for your help.

1 Like