[CVPR 2022] Eigenlanes: Data-Driven Lane Descriptors for Structurally Diverse Lanes
Dongkwon Jin, Wonhui Park, Seong-Gyun Jeong, Heeyeon Kwon, and Chang-Su Kim
Official implementation for "Eigenlanes: Data-Driven Lane Descriptors for Structurally Diverse Lanes" [paper] [supp] [video].
We construct a new dataset called "SDLane". SDLane is available at here. Now, only test set is provided due to privacy issues. All dataset will be provided soon.
Video
Related work
We wil also present another paper, "Eigencontours: Novel Contour Descriptors Based on Low-Rank Approximation", accepted to CVPR 2022 (oral) [github] [video].
Requirements
- PyTorch >= 1.6
- CUDA >= 10.0
- CuDNN >= 7.6.5
- python >= 3.6
Installation
- Download repository. We call this directory as
ROOT:
$ git clone https://github.com/dongkwonjin/Eigenlanes.git
- Download pre-trained model parameters and preprocessed data in
ROOT:
$ cd ROOT
$ unzip pretrained.zip
$ unzip preprocessed.zip
- Create conda environment:
$ conda create -n eigenlanes python=3.7 anaconda
$ conda activate eigenlanes
- Install dependencies:
$ conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
$ pip install -r requirements.txt
Directory structure
. # ROOT
├── Preprocessing # directory for data preprocessing
│ ├── culane # dataset name (culane, tusimple)
| | ├── P00 # preprocessing step 1
| | | ├── code
| | ├── P01 # preprocessing step 2
| | | ├── code
| │ └── ...
│ └── ... # etc.
├── Modeling # directory for modeling
│ ├── culane # dataset name (culane, tusimple)
| | ├── code
│ ├── tusimple
| | ├── code
│ └── ... # etc.
├── pretrained # pretrained model parameters
│ ├── culane
│ ├── tusimple
│ └── ... # etc.
├── preprocessed # preprocessed data
│ ├── culane # dataset name (culane, tusimple)
| | ├── P03
| | | ├── output
| | ├── P04
| | | ├── output
| │ └── ...
│ └── ...
.
Evaluation (for CULane)
To test on CULane, you need to install official CULane evaluation tools. The official metric implementation is available here. Please downloads the tools into ROOT/Modeling/culane/code/evaluation/culane/. The tools require OpenCV C++. Please follow here to install OpenCV C++. Then, you compile the evaluation tools. We recommend to see an installation guideline
$ cd ROOT/Modeling/culane/code/evaluation/culane/
$ make
Train
- Set the dataset you want to train (
DATASET_NAME) - Parse your dataset path into the
-dataset_dirargument. - Edit
config.pyif you want to control the training process in detail
$ cd ROOT/Modeling/DATASET_NAME/code/
$ python main.py --run_mode train --pre_dir ROOT/preprocessed/DATASET_NAME/ --dataset_dir /where/is/your/dataset/path/
Test
- Set the dataset you want to test (
DATASET_NAME) - Parse your dataset path into the
-dataset_dirargument. - If you want to get the performances of our work,
$ cd ROOT/Modeling/DATASET_NAME/code/
$ python main.py --run_mode test_paper --pre_dir ROOT/preprocessed/DATASET_NAME/ --paper_weight_dir ROOT/pretrained/DATASET_NAME/ --dataset_dir /where/is/your/dataset/path/
- If you want to evaluate a model you trained,
$ cd ROOT/Modeling/DATASET_NAME/code/
$ python main.py --run_mode test --pre_dir ROOT/preprocessed/DATASET_NAME/ --dataset_dir /where/is/your/dataset/path/
Preprocessing
Data preprocessing is divided into five steps, which are P00, P01, P02, P03, and P04. Below we describe each step in detail.
- In P00, the type of ground-truth lanes in a dataset is converted to pickle format.
- In P01, each lane in a training set is represented by 2D points sampled uniformly in the vertical direction.
- In P02, lane matrix is constructed and SVD is performed. Then, each lane is transformed to its coefficient vector.
- In P03, clustering is performed to obtain lane candidates.
- In P04, training labels are generated to train the SI module in the proposed SIIC-Net.
If you want to get the preproessed data, please run the preprocessing codes in order. Also, you can download the preprocessed data.
$ cd ROOT/Preprocessing/DATASET_NAME/PXX_each_preprocessing_step/code/
$ python main.py --dataset_dir /where/is/your/dataset/path/
Reference
@Inproceedings{
Jin2022eigenlanes,
title={Eigenlanes: Data-Driven Lane Descriptors for Structurally Diverse Lanes},
author={Jin, Dongkwon and Park, Wonhui and Jeong, Seong-Gyun and Kwon, Heeyeon and Kim, Chang-Su},
booktitle={CVPR},
year={2022}
}
503 Jan 4, 2023
58 Dec 6, 2022
32 Dec 15, 2022
21 Nov 23, 2022
74 Dec 15, 2022
28 Dec 13, 2022
69 Dec 8, 2022
58 Dec 23, 2022
129 Dec 24, 2022
78 Dec 10, 2022
100 Dec 19, 2022
190 Dec 27, 2022
219 Dec 28, 2022
55 Dec 16, 2022
63 Dec 16, 2022
71 Dec 28, 2022
104 Jan 5, 2023
239 Dec 26, 2022