about lane test

[thesun9999]

请问车道线预测是否可以生成掩码？我用culane预测了一下，只生成了.line.txt，请问怎样才能生成mask.png?

[voldemortX]

请问车道线预测是否可以生成掩码？我用culane预测了一下，只生成了.line.txt，请问怎样才能生成mask.png?

语义分割的方法应该可以出的。只是最终存储格式可能不是png。

[thesun9999]

感谢您的回答！请问是生成了掩码结果但是我没找到吗？/output中只看到了.txt的文件。预测的命令：python main_landet.py --test --config=/home/thesun/pytorch-auto-drive/configs/lane_detection/scnn/vgg16_culane.py --checkpoint=/home/thesun/pytorch-auto-drive/pt/vgg16_scnn_culane_20210309.pt

[voldemortX]

感谢您的回答！请问是生成了掩码结果但是我没找到吗？/output中只看到了.txt的文件。预测的命令：python main_landet.py --test --config=/home/thesun/pytorch-auto-drive/configs/lane_detection/scnn/vgg16_culane.py --checkpoint=/home/thesun/pytorch-auto-drive/pt/vgg16_scnn_culane_20210309.pt

应该是代码没有存，你看看可视化的功能呢

[thesun9999]

是的，可视化是可以选择可视化关键点和掩码的，所以我想应该是可以生成掩码的。请问是否有提供生成掩码的代码？我没有找到，谢谢！

[voldemortX]

是的，可视化是可以选择可视化关键点和掩码的，所以我想应该是可以生成掩码的。请问是否有提供生成掩码的代码？我没有找到，谢谢！

你是要对culane数据集生成图片吗？可视化可以推理一个文件夹的

[thesun9999]

是的，对culane数据集预测出车道线的掩码图片，但是目前output只生成了关键点的txt

[thesun9999]

您说的可视化可以推理一个文件夹，意思是可视化可以同时对culane数据集进行预测并且将得到的mask可视化出来吗？

[voldemortX]

您说的可视化可以推理一个文件夹，意思是可视化可以同时对culane数据集进行预测并且将得到的mask可视化出来吗？

对的。支持这种功能

[voldemortX]

@thesun9999 还有更高级的方式：https://www.github.com/voldemortX/pytorch-auto-drive/tree/master/docs%2Fadvanced%2FVISUALIZE_LANE_DATASETS.md

简单点就搞到一个文件夹里，跑文件夹可视化

[thesun9999]

感谢您的回复！我用可视化工具预测得到的车道线仍然是txt形式的，并没有mask生成

[voldemortX]

感谢您的回复！我用可视化工具预测得到的车道线仍然是txt形式的，并没有mask生成

如果style=line不是你想要的格式。你可能需要修改一下可视化或推理评测代码，直接存储网络后处理前的分割结果

[voldemortX]

@thesun9999 看来确实没有你要的格式。如果你能开发style=mask功能，欢迎pr贡献你的代码。