Merge pull request #58 from OpenDriveLab/dlc

OpenLane-V2 Benchmark V2.0

Author: faikit

.gitignore

@@ -138,7 +138,7 @@ data/OpenLane-V2/*
 !data/OpenLane-V2/data_dict_sample.json
 !data/OpenLane-V2/data_dict_example.json
 !data/OpenLane-V2/openlanev2.md5
-!data/OpenLane-V2/preprocess.py
+!data/OpenLane-V2/preprocess*
 !data/OpenLane-V2/data_dict_subset_A.json
 
 RoadData/vis

README-zh-hans.md

@@ -9,3 +9,5 @@ df62c1f6e6b3fb2a2a0868c78ab19c92  OpenLane-V2_subset_A_image_5.tar
 c73af4a7aef2692b96e4e00795120504  OpenLane-V2_subset_A_image_7.tar
 fb2f61e7309e0b48e2697e085a66a259  OpenLane-V2_subset_A_image_8.tar
 95bf28ccf22583d20434d75800be065d  OpenLane-V2_subset_A_info.tar
+de22c7be880b667f1b3373ff665aac2e  OpenLane-V2_subset_A_sdmap.tar
+f85321cfc387fe97417af5e61cec9d2f  OpenLane-V2_subset_A_info-ls.tar.gz

data/OpenLane-V2/openlanev2.md5

@@ -0,0 +1,45 @@
+# ==============================================================================
+# Binaries and/or source for the following packages or projects 
+# are presented under one or more of the following open source licenses:
+# preprocess.py    The OpenLane-V2 Dataset Authors    Apache License, Version 2.0
+#
+# Contact wanghuijie@pjlab.org.cn if you have any issue.
+#
+# Copyright (c) 2023 The OpenLane-v2 Dataset Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =============================================================================='
+
+from openlanev2.centerline.io import io
+from openlanev2.lanesegment.preprocessing import collect
+
+
+with_sd_map = False # TODO: include SD Maps as sensor inputs or not
+
+root_path = './OpenLane-V2'
+for file in io.os_listdir(root_path):
+    if file.endswith('json'):
+        subset = file.split('.')[0]
+        for split, segments in io.json_load(f'{root_path}/{file}').items():
+            collect(
+                root_path, 
+                {split: segments}, 
+                f'{subset}_{split}_ls' if not with_sd_map else f'{subset}_{split}_ls_sd',
+                with_sd_map = with_sd_map,
+                n_points={
+                    'area': 20,
+                    'centerline': 10,
+                    'left_laneline': 20,
+                    'right_laneline': 20,
+                },
+            )

data/OpenLane-V2/preprocess-ls.py

@@ -20,13 +20,22 @@
 # limitations under the License.
 # =============================================================================='
 
-from openlanev2.io import io
-from openlanev2.preprocessing import collect
+from openlanev2.centerline.io import io
+from openlanev2.centerline.preprocessing import collect
+
+
+with_sd_map = False # TODO: include SD Maps as sensor inputs or not
 
 root_path = './OpenLane-V2'
 for file in io.os_listdir(root_path):
     if file.endswith('json'):
         subset = file.split('.')[0]
         for split, segments in io.json_load(f'{root_path}/{file}').items():
             point_interval = 1 if split == 'train' else 20
-            collect(root_path, {split: segments}, f'{subset}_{split}', point_interval=point_interval)
+            collect(
+                root_path, 
+                {split: segments}, 
+                f'{subset}_{split}' if not with_sd_map else f'{subset}_{split}_sd', 
+                point_interval=point_interval,
+                with_sd_map=with_sd_map,
+            )