search

msracver / FCIS

Fully Convolutional Instance-aware Semantic Segmentation
MIT License
1.57k stars 415 forks source link

How to train fcis network with resnet-50 #94

Open sifengyang opened 6 years ago

sifengyang commented 6 years ago

Because my machine's gpu is gtx980, and its memory is 4 GB, so I can't train the fcis network by resnet-101 . there is the file (FCIS/fcis/symbols/resnet_v1_101_fcis.py), how can I modify the file?? please help me.

franciszzj commented 6 years ago 
 def get_resnet_50(self):
         sym = mx.sym.load('model/pretrained_model/resnet-50-symbol.json')
         return sym.get_internals()['data'], sym.get_internals()['relu1_output']

Hope this can help you.