search

zkyseu / PPlanedet

Lane detection with PaddlePaddle. PPLanedet contains many SOTA methods, e.g. CondLaneNet, SCNN, RESA, RTFormer,UFLD
MIT License
65 stars 8 forks source link

用Tusimple数据集训练不起来CLRNet #17

Open Qiuyanghang opened 1 month ago

Qiuyanghang commented 1 month ago

配置文件用的CLRNet中的resnet18_tusimple.py，出现下面报错，测试在Windows，包版本如下图

1 2 3 d83c45f223ba8cc634632315c6bb1e2
Qiuyanghang commented 1 month ago

我将Tusimple数据集格式整理成了下图格式，配置文件是没有进行改动的（想先测试能否用起来），望作者不吝赐教，十分感谢！

e369bb3f0a0db10b07f90b3c68abee9
zkyredstart commented 1 month ago

应该是车道线类别数没改，num_classes=7

zkyredstart commented 1 month ago

max_lanes = 7

zkyredstart commented 1 month ago

这个应该是配置文件的问题，我后续来修改一下，另外CLRNet建议在CULane上训练，Tusimple好像还没有训练。

Qiuyanghang commented 1 month ago

这个应该是配置文件的问题，我后续来修改一下，另外CLRNet建议在CULane上训练，Tusimple好像还没有训练。

刚修改训练起来了，好的后续我再在CULane训练测试，感谢!

Qiuyanghang commented 1 month ago

这个应该是配置文件的问题，我后续来修改一下，另外CLRNet建议在CULane上训练，Tusimple好像还没有训练。

刚修改训练起来了，好的后续我再在CULane训练测试，感谢! 这行权重个数我也添加成与类别数相同的了 https://github.com/zkyseu/PPlanedet/blob/e34a8014917b80a6886ff972bce88017d9431561/configs/clrnet/resnet18_tusimple.py#L25