Turoad/CLRNet

请问可以区分实线和虚线吗?

Shicc opened this issue · 7 comments

Shicc commented
请问可以区分实线和虚线吗?

未提供描述。

您好,我也有类似的疑问,请问您有解决方案了吗?

有了吗有了吗?

我有解决方案了,在网络的分类分支加个通道让原本的背景/车道线的二分类变成背景/实线/虚线的多分类就行。不过需要在加载数据的时候注意加入实线虚线的类别,然后clr_head里很多地方的索引值也要进行修改。具体来说,网络里原本是192个78维度的anchor向量,0、1维对应背景/车道线即分类分支的输出,2、3、4维度是起点的y、x、θ值、5维是线的长度length、6维之后是采样点的水平偏移量,2维之后都对应回归分支的输出,所以加上虚线实线的话在分类分支输出加个维度让它变成0、1、2维,剩下的维度往后顺移,让原本一共78维变成79维就可以了。

我有解决方案了,在网络的分类分支加个通道让原本的背景/车道线的二分类变成背景/实线/虚线的多分类就行。不过需要在加载数据的时候注意加入实线虚线的类别,然后clr_head里很多地方的索引值也要进行修改。具体来说,网络里原本是192个78维度的anchor向量,0、1维对应背景/车道线即分类分支的输出,2、3、4维度是起点的y、x、θ值、5维是线的长度length、6维之后是采样点的水平偏移量,2维之后都对应回归分支的输出,所以加上虚线实线的话在分类分支输出加个维度让它变成0、1、2维,剩下的维度往后顺移,让原本一共78维变成79维就可以了。

请问可以加个联系方式咨询下嘛