pytorch-SceneNetRGBD icon indicating copy to clipboard operation
pytorch-SceneNetRGBD copied to clipboard

Are there pretrained models on Scenenet only?

Open zeroAska opened this issue 6 years ago • 4 comments

Hi,

Is there a model only trained on Scenenet, but not on NYU/SUN dataset? Say if I wanted to use the pretrained weights on Scenenet and fine-tune on my own dataset, which pretrained weight shall I use on that google drive folder? nyu_models/rgb_scenenet_pretrain/converted_clean_model.pth or sun_models/rgb_scenenet_pretrain/converted_clean_model.pth?

Thanks!

zeroAska avatar Apr 23 '19 05:04 zeroAska

any should be fine.

ankurhanda avatar Apr 23 '19 07:04 ankurhanda

Thanks for replying! Are these two models already fine-tuned on nyu/sun respectively?

zeroAska avatar Apr 23 '19 17:04 zeroAska

I think so. I will dig up if I still have the scenenet trained models - they should be somewhere but these should be also usable for you.

ankurhanda avatar Apr 23 '19 17:04 ankurhanda

Great, thank you so much for the help! We are trying to use the scenenet's trajectories and the trained weight to build 3D semantic maps on its validation set. As scenenet has so many images, we don't have enough resources to train from scratch -- it will be awesome if we can directly use the trained models.

zeroAska avatar Apr 23 '19 17:04 zeroAska