search

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
824 stars 137 forks source link

predict on cityscapes/predict on different size #175

Open thesun9999 opened 4 months ago

thesun9999 commented 4 months ago

您好，我用下载的模型在cityscapes数据集上做预测的时候遇到一点问题，结果只显示一半的车道线，我猜想是因为输入尺寸变化了，请问要在哪里修改一下吗？ 命令：python tools/vis/lane_img_dir.py --image-path=/home/thesun/PointPainting-master/lineseg/test_data/05181432_02031.MP4 --image-suffix=.png --save-path=/home/thesun/PointPainting-master/lineseg/my_output/output_culane/driver_37_30frame/05181432_02031.MP4 --pred --config=/home/thesun/PointPainting-master/lineseg/configs/lane_detection/scnn/vgg16_culane.py --checkpoint=/home/thesun/PointPainting-master/lineseg/pt/vgg16_scnn_culane_20210309.pt --style=line 2024-03-08 20-37-09屏幕截图 2024-03-08 20-36-34屏幕截图

voldemortX commented 4 months ago

@thesun9999 推理应该是resize做的，这个可能和数据集的bias有关，比如训练数据某个高度以上没有车道线。你可以尝试把车道线数据集可视化正常的图片做成cityscapes的size，看看是否正常

voldemortX commented 4 months ago

如果也不正常，那确实可能是个可调整的bug

thesun9999 commented 4 months ago

@thesun9999 推理应该是resize做的，这个可能和数据集的bias有关，比如训练数据某个高度以上没有车道线。你可以尝试把车道线数据集可视化正常的图片做成cityscapes的size，看看是否正常

感谢回复！我先把cityscapes的图像resize一下看看能不能解决