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