JiayuanWang-JW / YOLOv8-multi-task

GNU Affero General Public License v3.0
247 stars 41 forks source link

关于分割头的设计? #93

Open Cxgmg opened 3 hours ago

Cxgmg commented 3 hours ago

请问作者为什么在该模型中一个分割类别就分配了一个分割头,是否可以按照检测头一样所有语义分割类就别只用一个分割头解决?

Cxgmg commented 3 hours ago

我认为这样可以有效降低模型参数量以及提速,但是检测精度可能会有所降低?不知道作者是否有进行相关尝试。

JiayuanWang-JW commented 2 hours ago

Hi @Cxgmg

你的想法非常好,如果分割头不是轻量级的,严重影响FPS的情况下,合二为一肯定更好,但是你也提到了弊端,精度会下降,由于我认为我设计的分割头并不是很臃肿,并且考虑到车道线和可行驶区域的特征有很大的偏差,所以才用了两个分割neck 和head。这样还可以最大程度上避免negative transfer。你也可以尝试把他们合二为一。