How to pretrain in imagenet1k?

VCIP-RGBD / DFormer

[ICLR 2024] DFormer: Rethinking RGBD Representation Learning for Semantic Segmentation

https://yinbow.github.io/Projects/DFormer/index.html

MIT License

142 stars 24 forks source link

How to pretrain in imagenet1k? #23

Closed ZivHong closed 3 months ago

ZivHong commented 3 months ago

Your work are awesome, the gratest part is pretrained in imagenet-1k that could provide a good basis for transfer learning in other task. However, I couldn't reproduce the training procedure only by Table 17 ,I'm wondering if you could provide the code for pretraining in imagenet-1k. Thank you for your response beforehand.

yinbow commented 3 months ago

Sure. Thanks for your attention to our work. For the pretrain code, please refer to https://github.com/VCIP-RGBD/RGBD-Pretrain .

ZivHong commented 3 months ago

I don't notice you had provide the information in readme, I apologize for my mistake. I appreciate your message, thank you a lot.

yinbow commented 3 months ago

Never mind 😃 . If you have any other questions about our work, feel free to contact me.