README.md 1.84 KB
Newer Older
Chengyu Wang's avatar
Chengyu Wang committed
1
## InternImage-based Baseline for CVPR23 OpenLane-V2 Challenge!!!!
Charlie W's avatar
Charlie W committed
2

Chengyu Wang's avatar
Chengyu Wang committed
3
We improve our baseline with a more powerful image backbone: **InaternImage**, which shows its excellent ability within a series of leaderboards and benchmarks, such as *COCO* and *nuScenes*.
Charlie W's avatar
Charlie W committed
4
5


Chengyu Wang's avatar
Chengyu Wang committed
6
7
8
9
10
11
12
13
14
#### 1. Requirements
```bash
python>=3.8
torch==1.11
mmcv-full>=1.5.2
mmdet==2.28.0
mmsegmentation==0.29.1
timm
```
Charlie W's avatar
Charlie W committed
15
16


Chengyu Wang's avatar
Chengyu Wang committed
17
18
19
20
21
### 2. Install DCNv3 for InternImage
```bash
cd plugin/mmdet3d/baseline/models/backbones/ops_dcnv3
bash make.sh # requires torch>=1.10
```
Charlie W's avatar
Charlie W committed
22

Chengyu Wang's avatar
Chengyu Wang committed
23
### 3. Train with InternImage-Small
Charlie W's avatar
Charlie W committed
24

Chengyu Wang's avatar
Chengyu Wang committed
25
26
27
```bash
./tools/dist_train.sh plugin/mmdet3d/configs/internimage-s.py 8
```
Charlie W's avatar
Charlie W committed
28

Chengyu Wang's avatar
Chengyu Wang committed
29
Notes: InternImage provides abundant pre-trained model weights that can be used!!!
Charlie W's avatar
Charlie W committed
30
31


Chengyu Wang's avatar
Chengyu Wang committed
32
### 4. Performance compared to baseline
Charlie W's avatar
Charlie W committed
33

Chengyu Wang's avatar
Chengyu Wang committed
34
35
36
37
|             | OpenLane-V2 Score | DET<sub>l</sub> | DET<sub>t</sub> | TOP<sub>ll</sub> | TOP<sub>lt</sub> | F-Score |
|-------------|-------------------|-----------------|-----------------|------------------|------------------|---------|
| base r50    | 0.292             | 0.183           | 0.457           | 0.022            | 0.143            | 0.215   |
| InternImage | 0.325             | 0.194           | 0.537           | 0.02             | 0.17             | 0.224   |
Charlie W's avatar
Charlie W committed
38
39


Chengyu Wang's avatar
Chengyu Wang committed
40
41
42
43
44
45
## Challenge Timeline
- Pending - Challenge Period Open.
- Jun 01, 2023 - Challenge Period End.
- Jun 03, 2023 - Finalist Notification.
- Jun 10, 2023 - Technical Report Deadline.
- Jun 12, 2023 - Winner Announcement.
Charlie W's avatar
Charlie W committed
46
47
48
<p align="right">(<a href="#top">back to top</a>)</p>


Chengyu Wang's avatar
Chengyu Wang committed
49
50
## Leaderboard 
To be released.
Charlie W's avatar
Charlie W committed
51
52
53
54

<p align="right">(<a href="#top">back to top</a>)</p>

## License
Chengyu Wang's avatar
Chengyu Wang committed
55
Before using the dataset, you should register on the website and agree to the terms of use of the [nuScenes](https://www.nuscenes.org/nuscenes).
Charlie W's avatar
Charlie W committed
56
57
58
All code within this repository is under [Apache License 2.0](./LICENSE).

<p align="right">(<a href="#top">back to top</a>)</p>