deeplab v3+模型训练

PaddlePaddle / models

Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.

Apache License 2.0

6.9k stars 2.91k forks source link

deeplab v3+模型训练 #1708

Open xieqingxing opened 5 years ago

xieqingxing commented 5 years ago

我使用官方给出的训练好的模型,在cityspace的验证集上eval,不能做到0.7+的miou,大概只有0.28.使用预训练模型进行训练在几万轮训练后,精度上升也不明显. 另外想问下,在paddle中,如何设置各个类别的softmax_with_cross_entropy的权重,因为各个类别像素数量不是平均.

cjld commented 5 years ago

如果您是使用batch norm的话，请保证batch size至少是8，如果显存不足，您可以尝试使用group norm，在最新版本的models里面已经更新了相关的参数

https://github.com/PaddlePaddle/models/blob/develop/fluid/PaddleCV/deeplabv3+/README.md#预训练模型准备

Cugtyt commented 5 years ago

@cjld 请问在Pascal VOC上从随机权重开始训练需要多少轮可以复现89.0mIOU，有相关的日志可以参考吗

eddieheyutong commented 4 years ago

您好请问您方便留个联系方式嘛，同样在用deeplabv3+有些问题想请教您一下