search

midasklr / PPLiteSeg.pytorch

pytorch of the SOTA real-time segmentation network ppliteseg
20 stars 2 forks source link

训练 #1

Open ywj55555 opened 2 years ago

ywj55555 commented 2 years ago

请问一下，三个输出是怎么训练的呀，是直接计算三个损失之和吗？请问可以参考一下您的训练部分的代码吗

midasklr commented 2 years ago

对，分别计算三个分割head(1/8,1/16and1/32)的损失，相加（系数为1）作为总的loss，然后反向传播更新网络参数，可以直接参考ddrnet的训练代码。

VikingKang commented 2 years ago

预测呢，用的平均还是单独的哪一层

midasklr commented 2 years ago

预测是1/8尺度的输出