ZJULearning / RMI

This is the code for the NeurIPS 2019 paper Region Mutual Information Loss for Semantic Segmentation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Not available to download ResNet101 pretrained model.

koda12344505 opened this issue · comments

Hello there,
I tried to download ResNet101 pretrained model using by model_store.py scripts,

but, it was not possible to download ResNet101 pretrained model.

Can you provide pretrained model by other way?
otherwise, it is not possible to reproduce your work....
i just trained from scratch using camvid datasets... it was around 60% mIoU using RMI Loss..

thanks.

by the way, i just wonder why some implementations are different with paper?
for example,

rmi_pool_size is 4 in train.sh
but in paper, 3x3 produce best performance...

thanks..

@koda12344505

  1. The pre-trained model is downloaded from https://hangzhang.org/PyTorch-Encoding/model_zoo/imagenet.html. You may check it, I will also try to load the pre-trained model in this repo.

  2. rmi_pool_size=4 (DF in the paper) is the size of the pooling kernel (also the stride) of the max/avg pooling. It is also the default setting (with rmi_radius=3, R in the paper) in most tables. It is not the best setting with the best performance, but it can work with less extra memory cost than the best setting.

Please check https://github.com/ZJULearning/RMI/releases/tag/1.0.0 for pre-trained models.

Thank you for your quick update :)

I will try to train DeepLabv3+ with resnet101 model on camvid

Thank you so much again