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

1

error1
error1.png706×532 24.5 KB

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?

2

Hi @DarkWings,

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

1 Like
3

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.

4

This works for you?

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

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

1 Like