TUI-NICR / ESANet

ESANet: Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis
Other
236 stars 51 forks source link

Pretrained weight for Cityscapes (RGB-only) #53

Closed matteosodano closed 1 year ago

matteosodano commented 1 year ago

Hi, thanks for the great work! In the paper, you mention you trained a single-modality RGB version of ESANet on Cityscapes. Do you happen to still have the weights, by any chance? Also, was it still the ResNet34+NBt1D trained with the same procedure you specify?

Thanks again!

danielS91 commented 1 year ago

Unfortunately, we do not have the weights of this run anymore. However, you can follow the instructions given here for Cityscapes: https://github.com/TUI-NICR/ESANet#training

Just change "--modality rgbd" to "--modality rgb". The remaining hyperparameter must be similar.

matteosodano commented 1 year ago

Ok, thanks!