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
847 stars 138 forks source link

lstr demo #30

Closed lucasjinreal closed 2 years ago

lucasjinreal commented 3 years ago

does there a demo visualize of lstr model on tusimple or culane?

voldemortX commented 3 years ago

@jinfagang Yes, there is. For example on a video file:

python visualize_lane.py --image-path=tusimple_val_1min.avi --save-path=test_tusimple.avi --method=lstr --backbone=resnet18s --dataset=tusimple --continue-from=<weight file path> --height=360 --width=640

More details in VISUALIZATION.md

p.s. Thanks for reminding me, there is a bug in LSTR visualization, which is now fixed on master.

voldemortX commented 2 years ago

Bug fixed. Feel free to reopen!