pytorch-基于Pytorch实现的LaneNet算法.zip

## Installation `python setup.py install` ## Usage ### Train To train on the test data included in the repo, `python3 lanenet/train.py --dataset ./data/training_data_example` #### TUSimple dataset Download TUsimple dataset When done run the script in the `scripts`-folder `python tusimple_transform.py --src_dir <directory of downloaded tusimple>` After this run training as before: `python3 lanenet/train.py --dataset <tusimple_transform script output folder>` #### Custom dataset To train on a custom dataset, the easiest approach is to make sure it follows the format laid out in the data folder. Alternatively write a custom PyTorch dataset class (if you do, feel free to provide a PR)