voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help
BSD 3-Clause "New" or "Revised" License
837 stars 137 forks source link

测试culane数据集的demo #109

Closed GongDianCheng closed 2 years ago

GongDianCheng commented 2 years ago

感谢您的开源,被文章中提出的贝塞尔曲线拟合吸引过来的。在尝试测试culane数据集时,执行如下脚本: python main_landet.py --test --config=configs/lane_detection/bezierlanenet/resnet34_culane_aug1b.py --checkpoint=resnet34_bezierlanenet_culane_aug1b_20211109.pt 然而得到的输出是test_cfg.json,并不是数据集的测试结果, 请问我的脚本有问题吗,我是不是忽略了哪些关键点

voldemortX commented 2 years ago

@GongDianCheng 可以看一下 https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/LANEDETECTION.md

如果要测数据集的F-1结果,需要autotest_culane.sh 如果要预测的*lines.txt文件的话,它们会出现在缓存文件夹./output

GongDianCheng commented 2 years ago

@GongDianCheng 可以看一下 https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/LANEDETECTION.md

如果要测数据集的F-1结果,需要autotest_culane.sh 如果要预测的*lines.txt文件的话,它们会出现在缓存文件夹./output

您好,感谢您的回复! 我注意到了有一个步骤mkdir output ,我创建了该文件夹,但是运行那个问题所述命令后,output下没有结果,这也是我疑惑的地方

voldemortX commented 2 years ago

这有点奇怪,你运行main_landet.py的时候有输出和报错信息么。

GongDianCheng commented 2 years ago

这有点奇怪,你运行main_landet.py的时候有输出和报错信息么。

(torch1.9) gdc@gdc:~/gong/lane-detection/pytorch-auto-drive-master$ python main_landet.py --test --config=configs/lane_detection/bezierlanenet/resnet34_culane_aug1b.py --checkpoint=resnet34_bezierlanenet_culane_aug1b_20211109.pt Successfully complied line nms for LaneATT. Loaded torchvision ImageNet pre-trained weights V1. cuda:0 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9/9 [00:01<00:00, 4.92it/s] Files saved at: ./checkpoints/resnet34_bezierlanenet_culane-aug2. Tensorboard log at: ./checkpoints/tb_logs/resnet34_bezierlanenet_culane-aug2

GongDianCheng commented 2 years ago

2022-07-13 10-40-18 的屏幕截图

voldemortX commented 2 years ago

@GongDianCheng 你的测试是几秒钟就完成了么,测试集应该有上万张图,确认一下数据集有没有问题?

GongDianCheng commented 2 years ago

是这样的,我修改了culane的test.list,只包含了几张图片,想看一下论文中的测试效果

voldemortX commented 2 years ago

https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/datasets/CULANE.md

voldemortX commented 2 years ago

是这样的,我修改了culane的test.list,只包含了几张图片,想看一下论文中的测试效果

额,那修改前的测试也有这个问题么

voldemortX commented 2 years ago

@GongDianCheng 如果只想看几张图的效果的话,可以放到一个文件夹里直接跑可视化: https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/VISUALIZATION.md#lane-points-image-folder

GongDianCheng commented 2 years ago

@voldemortX 我在测试以下,感谢回复

GongDianCheng commented 2 years ago

感谢您提供的idea,问题已解决,应该是我数据集的问题。这是一个伟大全面的自动驾驶框架,感谢开源精神!

voldemortX commented 2 years ago

感谢您提供的idea,问题已解决,应该是我数据集的问题。这是一个伟大全面的自动驾驶框架,感谢开源精神!

群众的支持是我们前进的动力!

Hzy98 commented 1 year ago

你好,我想获得测试数据集的f1结果,并且按照CULANE.md中准备python评估脚本,autotest_culane.sh也改为了python。eval.sh和eval_validation.sh中data_dir也都已修改了。但是在执行:./autotest_culane.sh resnet18_bezierlanenet_culane-aug1b test checkpoints还是报错了: image 我的output文件夹中包含测试预测的车道线点: image image

voldemortX commented 1 year ago

@Hzy98 看起来是opencv问题,你是开了python后端了吗?

autotest_culane.sh文件发一下?感觉你没进python后端。要是还有问题可以开个新的issue

Hzy98 commented 1 year ago

image 根据CULANE.md里说的,我只是把autotest_culane.sh中backend修改为了python,别的没有改动。 image

voldemortX commented 1 year ago

image 根据CULANE.md里说的,我只是把autotest_culane.sh中backend修改为了python,别的没有改动。 image

@Hzy98 最顶上改了就行,if里面别改呀。。。

注意你要改路径的eval.sh等也是culane_evaluation_py里面的。

Hzy98 commented 1 year ago

image 根据CULANE.md里说的,我只是把autotest_culane.sh中backend修改为了python,别的没有改动。 image

@Hzy98 最顶上改了就行,if里面别改呀。。。

哦哦 我都忘了改了if里面的了,以为只改了我圈出来的地方。谢谢。