bubbliiiing / segformer-pytorch

这是一个segformer-pytorch的源码,可以用于训练自己的模型。
MIT License
271 stars 35 forks source link

请问pascal voc是加载官方预训练跑出来的? #10

Open joe3154 opened 1 year ago

joe3154 commented 1 year ago

我自己跑官方的预训练backbone效果很差。

bubbliiiing commented 1 year ago

是的,pretrained=True吗

worldlyesterday commented 1 year ago

我也难以复现原作中ADE20k数据集的mIoU,可能需要调整超参数?

joe3154 commented 1 year ago

我参数配置调的都是一样的,用的ADE预训练权重,但是跑voc很差

worldlyesterday commented 1 year ago

请问一下你是怎么载入官方的预训练权重呀,我之前试了一下匹配不上

joe3154 commented 1 year ago

dict按key加载

joe3154 commented 1 year ago

不冻住backbone效果好差

bubbliiiing commented 1 year ago

0 0不应该啊,你们是怎么操作的,我就是正常的pretrained=True开始的 不是官方那个开始的

bubbliiiing commented 1 year ago

你们的训练参数啥是什么

joe3154 commented 1 year ago

除了LR_Scheduler不是poly,都和官方一样

joe3154 commented 1 year ago

我加载你voc的b2可以跑到78.38%,只加载imagenet跑50epoch一半都不到。

worldlyesterday commented 1 year ago

我加载voc的b2预训练权重在ADE20k上训练,和原论文的实验结果也有挺大差距的

joe3154 commented 1 year ago

非常感谢! @bubbliiiing ,没有问题了。

kimer1 commented 1 year ago

请问怎么解决跑ADE20k数据集效果很差的问题