您好，我想要修改网络结构，并且重新预训练，可以提供相关代码吗？

bubbliiiing / segformer-pytorch

这是一个segformer-pytorch的源码，可以用于训练自己的模型。

MIT License

282 stars 36 forks source link

Open waw123456 opened 3 weeks ago

waw123456 commented 3 weeks ago

请问预训练使用的数据集是什么，是ImageNet-1K吗？您的预训练权重貌似是比官网的预训练权重效果更好的，可否提供预训练的代码，可否告知预训练的细节？我想对原Segformer进行一些主干网络的魔改，谢谢！