How to use other models for other projects??

[Uehwan]

The notebook demo contains the following part:

net_encoder = ModelBuilder.build_encoder( arch='resnet50dilated', fc_dim=2048, weights='ckpt/ade20k-resnet50dilated-ppm_deepsup/encoder_epoch_20.pth') net_decoder = ModelBuilder.build_decoder( arch='ppm_deepsup', fc_dim=2048, num_class=150, weights='ckpt/ade20k-resnet50dilated-ppm_deepsup/decoder_epoch_20.pth', use_softmax=True)

I'm wondering how I can use other networks such as HRNet. The tricky part is how to specify fc_dim, etc. Plus, the code in "from mit_semseg.models import ModelBuilder, SegmentationModule" does not contain the code for defining the decoder part of HRNet.

Thank you in advance.

[GevHovh]

fc_dim for HRNet is 720 decoder part for HRNet is 'C1'

So yor code for HRNet should be as follows:

net_encoder = ModelBuilder.build_encoder( arch='hrnetv2', fc_dim=720, weights=''ckpt/ade20k-hrnetv2-c1/encoder_epoch_30.pth') net_decoder = ModelBuilder.build_decoder( arch='c1', fc_dim=720, num_class=150, weights=''ckpt/ade20k-hrnetv2-c1/decoder_epoch_30.pth', use_softmax=True)