MXNet Forum

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


#1


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.


#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.