search

Ze-Yang / Lane-Detection-with-ERFNet

Paddle Implementation for 2020 China Hualu Cup Data Lake Algorithm Competition co-hosted with Baidu AI (1st Place for Lane Detection Track) 2020中国华录杯·数据湖算法大赛(百度AI协办)—车道线识别赛道
Apache License 2.0
63 stars 15 forks source link

请问如何对预测输出的车道线添加与标注一样的颜色呢 #2

Open liu4791959 opened 3 years ago

liu4791959 commented 3 years ago

预测输出的车道线是单一的白色，这样不便展示，希望能和数据集的标注一样有对应的颜色，以体现不同的种类。请问有什么好的方法嘛

ascetic-monk commented 3 years ago

您好！ 首先跟您介绍一下本次比赛的标签，本次比赛的标签是.png形式的灰度图，所以不存在您说的和数据标注一样颜色的问题，我们的输出和label的形式是一样的。 按照比赛的规定，我们的输出是灰度图，灰度0~19对应相应的label，所以输出的是您所说的白色，灰度图可以通过伪彩图转换进行可视化，采用matplotlib和cv2的相关工具即可，也可以采用PaddleSeg提供的相关工具。

Ze-Yang commented 3 years ago

We have added visualization feature in utils/visualization.py, hope that it helps. Thanks.