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

visualize the gts only File: lane_img_dir.py #122

Closed shifan-Z closed 1 year ago

shifan-Z commented 1 year ago

Thanks for posting your excellent code! I want to visualize the gts of dataset CULane. To use the code, that it seems it is for the inference model. So if I just wanna show the gts, which config file should I provide? It seems the config file must exist. https://github.com/voldemortX/pytorch-auto-drive/blob/8cc1d30c9aa56ffb947f0a96eff48d8a4175d63c/tools/vis/lane_img_dir.py#L18

voldemortX commented 1 year ago

@shifan-Z Any config that uses CULane should suffice. While we only support image folder inputs. So if you want to parse the entire GT structure from a certain dataset, you may refer to advanced/VISUALIZE_LANE_DATASETS.md.

For CULane, you can try this config

shifan-Z commented 1 year ago

Thank you so much!